Searched refs:graph_transfer_info (Results 1 – 9 of 9) sorted by relevance
60 const string& name, GraphTransferInfo* graph_transfer_info) { in FindNodeInfo() argument62 *graph_transfer_info->mutable_node_info()) { in FindNodeInfo()139 GraphTransferInfo& graph_transfer_info = in SetupGraph() local144 graph_transfer_info.graph_input_node_info()) { in SetupGraph()146 FindNodeInfo(graph_input.name(), &graph_transfer_info); in SetupGraph()152 CHECK_EQ(graph_transfer_info.graph_output_node_info_size(), 1); in SetupGraph()154 graph_transfer_info.graph_output_node_info()) { in SetupGraph()155 const int new_output_node_id = graph_transfer_info.node_info_size() + in SetupGraph()156 graph_transfer_info.const_node_info_size() + in SetupGraph()160 *graph_transfer_info.add_node_info(); in SetupGraph()[all …]
211 const GraphTransferInfo& graph_transfer_info) { in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo() argument215 graph_transfer_info.graph_input_node_info()) { in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()227 graph_transfer_info.graph_output_node_info()) { in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()239 graph_transfer_info.SerializeAsString()); in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
71 const string& name, GraphTransferInfo* graph_transfer_info);
10 tensorflow/core/framework/graph_transfer_info.pb_text.cc
10 tensorflow/core/framework/graph_transfer_info.pb.cc
14 tensorflow/core/framework/graph_transfer_info.proto
10 tensorflow/core/framework/graph_transfer_info.pb.h
149 "tensorflow/core/framework/graph_transfer_info.proto"
197 "framework/graph_transfer_info.proto",