Searched refs:control_ret_nodes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 188 absl::flat_hash_set<Node*>* control_ret_nodes);
203 absl::flat_hash_set<Node*>* control_ret_nodes);
224 absl::flat_hash_set<Node*>* control_ret_nodes);
505 absl::flat_hash_set<Node*>* control_ret_nodes) { in GetControlRetNodes() argument
512 control_ret_nodes->insert(node_it->second); in GetControlRetNodes()
521 absl::flat_hash_set<Node*>* control_ret_nodes) { in Convert() argument
711 exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes)); in Convert()
728 absl::flat_hash_set<Node*> control_ret_nodes; in ConvertLibFunction() local
731 &control_ret_nodes)); in ConvertLibFunction()
733 return control_ret_nodes.contains(n) in ConvertLibFunction()
[all …]
|
| D | export_graphdef.h | 45 absl::flat_hash_set<Node*>* control_ret_nodes);
|
| D | import_model.cc | 197 absl::InlinedVector<Node*, 4>* control_ret_nodes,
211 const absl::InlinedVector<Node*, 4>& control_ret_nodes,
327 const absl::InlinedVector<Node*, 4>& control_ret_nodes);
1032 absl::InlinedVector<Node*, 4>* control_ret_nodes, in GetArgsAndRetsFromFunctionBody() argument
1046 *control_ret_nodes = fbody.control_ret_nodes; in GetArgsAndRetsFromFunctionBody()
1138 absl::InlinedVector<Node*, 4> control_ret_nodes; in ConvertLibFunction() local
1141 &control_ret_nodes, &resource_arg_unique_ids); in ConvertLibFunction()
1144 mlir_func_name, func_type, arg_nodes, ret_nodes, control_ret_nodes, in ConvertLibFunction()
1160 const absl::InlinedVector<Node*, 4>& control_ret_nodes, in Convert() argument
1188 ret_nodes, control_ret_nodes)); in Convert()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.h | 84 gtl::InlinedVector<Node*, 4> control_ret_nodes; member
|
| D | function.cc | 899 std::unordered_set<StringPiece, StringPieceHasher> control_ret_nodes; in PruneFunctionBody() local
901 control_ret_nodes.insert(control_ret.second); in PruneFunctionBody()
915 (control_ret_nodes.find(n->name()) != control_ret_nodes.end())) { in PruneFunctionBody()
2138 for (Node* fbody_node : fbody->control_ret_nodes) { in InlineFunctionBody()
2373 this->control_ret_nodes.reserve(control_ret_node_names.size()); in FunctionBody()
2376 this->control_ret_nodes.push_back(n); in FunctionBody()
|
| D | process_function_library_runtime.cc | 581 control_ret_node_names->reserve(fbody->control_ret_nodes.size()); in GetGraphAndArgRets()
582 for (const Node* node : fbody->control_ret_nodes) { in GetGraphAndArgRets()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 889 control_sources = {fbody.control_ret_nodes.begin(), in ValidateSideEffectsExecution()
890 fbody.control_ret_nodes.end()}; in ValidateSideEffectsExecution()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.cc | 332 for (const auto& control_ret_node : fbody->control_ret_nodes) { in HashFunction()
|