Searched refs:border_inputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform.cc | 172 const std::vector<string> border_inputs = in FuseRemoteGraph() local
176 for (size_t i = 0; i < border_inputs.size(); ++i) { in FuseRemoteGraph()
177 VLOG(2) << "Border Input(" << i << "): " << border_inputs.at(i); in FuseRemoteGraph()
184 border_inputs, border_outputs, remote_graph_executor_name, in FuseRemoteGraph()
238 const std::vector<string> border_inputs = in PlaceRemoteGraphArguments() local
248 inputs, outputs, fused_node_names, border_inputs, border_outputs, in PlaceRemoteGraphArguments()
|
| D | remote_fused_graph_execute_utils.cc | 767 std::vector<string>& border_inputs = std::get<1>(ci); in ClusterizeNodes() local
783 if (std::find(border_inputs.begin(), border_inputs.end(), src_name) == in ClusterizeNodes()
784 border_inputs.end()) { in ClusterizeNodes()
785 border_inputs.emplace_back(src_name); in ClusterizeNodes()
886 const std::vector<string>& border_inputs, in BuildClusterByBorder() argument
916 for (const string& input : border_inputs) { in BuildClusterByBorder()
937 std::get<1>(*cluster) = border_inputs; in BuildClusterByBorder()
953 const std::vector<string>& border_inputs = std::get<1>(cluster); in FuseCluster() local
968 border_inputs, border_outputs, require_shape_type, &graph, &fused_node)); in FuseCluster()
1060 const std::vector<string>& border_inputs, in FuseRemoteGraphByBorder() argument
[all …]
|
| D | remote_fused_graph_execute_utils.h | 208 static Status BuildClusterByBorder(const std::vector<string>& border_inputs,
242 const std::vector<string>& border_inputs,
260 const std::vector<string>& border_inputs,
|