DirectionSession涉及的重要的数据类型

struct PerPartitionExecutorsAndLib {
Graph* graph = nullptr;                  // not owned.
Device* device = nullptr;                // not owned.
FunctionLibraryRuntime* flib = nullptr;  // not owned.
std::unique_ptr executor;

};
struct ExecutorsAndKeys {
ExecutorsAndKeys() : step_count(0) {}

std::atomic_int_fast64_t step_count;
std::unique_ptr graph;
NameNodeMap name_to_node;
std::vector items;
std::unordered_map input_name_to_index;
std::unordered_map input_name_to_rendezvous_key;
std::unordered_map output_name_to_index;
std::unordered_map output_name_to_rendezvous_key;

DataTypeVector input_types;
DataTypeVector output_types;

CallableOptions callable_options;

int64 collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey;

};

你可能感兴趣的:(tensorflow源代码阅读)