Tensorflow研究学习
星空探索
linux
展开
-
Tensorflow之class Tensor
class Tensor { public: Tensor(); Tensor(DataType type, const TensorShape& shape); Tensor(Allocator* a, DataType type, const TensorShape& shape); Tensor(Allocator* a, DataType type, c原创 2017-09-11 21:25:31 · 1727 阅读 · 0 评论 -
Tensorflow之class Allocator
class Allocator { public: #ifdef EIGEN_VECTORIZE_AVX512 static constexpr size_t kAllocatorAlignment = 64; #else static constexpr size_t kAllocatorAlignment = 32; #endif virtual ~Allo原创 2017-09-11 21:33:44 · 554 阅读 · 0 评论 -
线程池ThreadPool
tensorflow::tensorflow/core/lib/core/threadpool.h namespace tensorflow { namespace thread { class ThreadPool { public: ThreadPool(Env* env, const ThreadOptions& thread_options, const原创 2017-12-18 09:15:29 · 280 阅读 · 0 评论 -
tf中graph定义
原生定义: tensorflow::tensorflow/core/framework/graph.proto message GraphDef { repeated NodeDef node = 1; VersionDef versions = 4; int32 version = 3 [deprecated = true]; FunctionDefLibra原创 2017-12-18 09:49:42 · 966 阅读 · 0 评论 -
tf中graph简要实现
GraphDef::GraphDef() : ::google::protobuf::Message(), _internal_metadata_(NULL) { if (GOOGLE_PREDICT_TRUE(this != internal_default_instance())) { protobuf_tensorflow_2fcore_2fframework_2fgr原创 2017-12-18 09:56:25 · 456 阅读 · 0 评论 -
tf内部图描述graph
// Thread compatible but not thread safe. class Graph { public: explicit Graph(const OpRegistryInterface* registry); explicit Graph(const FunctionLibraryDefinition& flib_def); ~Graph();原创 2017-12-18 10:05:25 · 509 阅读 · 0 评论