Searched refs:border_outputs (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | remote_fused_graph_rewriter_transform.cc | 180 const std::vector<string> border_outputs = in FuseRemoteGraph() local 185 for (size_t i = 0; i < border_outputs.size(); ++i) { in FuseRemoteGraph() 186 VLOG(2) << "Border Output(" << i << "): " << border_outputs.at(i); in FuseRemoteGraph() 190 border_inputs, border_outputs, remote_graph_executor_name, in FuseRemoteGraph() 246 const std::vector<string> border_outputs = in PlaceRemoteGraphArguments() local 254 inputs, outputs, fused_node_names, border_inputs, border_outputs, in PlaceRemoteGraphArguments()
|
D | remote_fused_graph_execute_utils.cc | 769 std::vector<string>& border_outputs = std::get<2>(ci); in ClusterizeNodes() local 811 if (std::find(border_outputs.begin(), border_outputs.end(), in ClusterizeNodes() 812 identity_dst_name) == border_outputs.end()) { in ClusterizeNodes() 813 border_outputs.emplace_back(identity_dst_name); in ClusterizeNodes() 816 if (std::find(border_outputs.begin(), border_outputs.end(), in ClusterizeNodes() 817 dst_name) == border_outputs.end()) { in ClusterizeNodes() 818 border_outputs.emplace_back(dst_name); in ClusterizeNodes() 889 const std::vector<string>& border_outputs, const GraphDef& graph_def, in BuildClusterByBorder() argument 897 for (const string& output : border_outputs) { in BuildClusterByBorder() 940 std::get<2>(*cluster) = border_outputs; in BuildClusterByBorder() [all …]
|
D | remote_fused_graph_execute_utils.h | 209 const std::vector<string>& border_outputs, 243 const std::vector<string>& border_outputs, 261 const std::vector<string>& border_outputs,
|