Searched refs:remote_graph_executor_name (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform.cc | 71 string* remote_graph_executor_name) { in ParseArguments() argument
96 "", remote_graph_executor_name)); in ParseArguments()
101 CHECK(!remote_graph_executor_name->empty()); in ParseArguments()
148 string remote_graph_executor_name; in FuseRemoteGraph() local
153 &remote_graph_executor_name)); in FuseRemoteGraph()
169 fused_node_names, remote_graph_executor_name, require_shape_type, in FuseRemoteGraph()
184 border_inputs, border_outputs, remote_graph_executor_name, in FuseRemoteGraph()
193 fused_op_types, remote_graph_executor_name, require_shape_type, in FuseRemoteGraph()
197 mutable_input_graph_def, inputs, outputs, remote_graph_executor_name, in FuseRemoteGraph()
222 string remote_graph_executor_name; in PlaceRemoteGraphArguments() local
[all …]
|
| D | remote_fused_graph_execute_utils.h | 224 const string& remote_graph_executor_name,
244 const string& remote_graph_executor_name, const bool require_shape_type,
264 const string& remote_graph_executor_name, GraphDef* graph_def);
|
| D | remote_fused_graph_execute_utils.cc | 946 const string& remote_graph_executor_name, const bool require_shape_type, in FuseCluster() argument
951 CHECK(!remote_graph_executor_name.empty()); in FuseCluster()
967 remote_fused_graph_node_name, remote_graph_executor_name, subgraph_def, in FuseCluster()
1062 const string& remote_graph_executor_name, const bool require_shape_type, in FuseRemoteGraphByBorder() argument
1070 remote_graph_executor_name, require_shape_type, output_graph_def); in FuseRemoteGraphByBorder()
1116 const string& remote_graph_executor_name, GraphDef* graph_def) { in PlaceRemoteGraphArguments() argument
1129 remote_graph_executor_name, in PlaceRemoteGraphArguments()
1181 string remote_graph_executor_name; in FuseRemoteGraphByPlacedArguments() local
1213 remote_graph_executor_name = attr.at(3); in FuseRemoteGraphByPlacedArguments()
1217 if (GetExecutorBuildFunc(remote_graph_executor_name) == nullptr) { in FuseRemoteGraphByPlacedArguments()
[all …]
|