Home
last modified time | relevance | path

Searched refs:control_ret_nodes (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc158 absl::flat_hash_set<Node*>* control_ret_nodes);
173 absl::flat_hash_set<Node*>* control_ret_nodes);
194 absl::flat_hash_set<Node*>* control_ret_nodes);
438 absl::flat_hash_set<Node*>* control_ret_nodes) { in GetControlRetNodes() argument
445 control_ret_nodes->insert(node_it->second); in GetControlRetNodes()
454 absl::flat_hash_set<Node*>* control_ret_nodes) { in Convert() argument
608 exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes)); in Convert()
625 absl::flat_hash_set<Node*> control_ret_nodes; in ConvertLibFunction() local
628 &control_ret_nodes)); in ConvertLibFunction()
630 return control_ret_nodes.contains(n) in ConvertLibFunction()
[all …]
Dexport_graphdef.h44 absl::flat_hash_set<Node*>* control_ret_nodes);
Dimport_model.cc220 absl::InlinedVector<Node*, 4>* control_ret_nodes);
232 const absl::InlinedVector<Node*, 4>& control_ret_nodes,
353 const absl::InlinedVector<Node*, 4>& control_ret_nodes);
1217 absl::InlinedVector<Node*, 4>* control_ret_nodes) { in GetArgsAndRetsFromFunctionBody() argument
1226 *control_ret_nodes = fbody.control_ret_nodes; in GetArgsAndRetsFromFunctionBody()
1319 absl::InlinedVector<Node*, 4> control_ret_nodes; in ConvertLibFunction() local
1321 &control_ret_nodes); in ConvertLibFunction()
1324 mlir_func_name, func_type, arg_nodes, ret_nodes, control_ret_nodes, in ConvertLibFunction()
1427 const absl::InlinedVector<Node*, 4>& control_ret_nodes, in Convert() argument
1453 ret_nodes, control_ret_nodes)); in Convert()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_graph_optimization_pass.cc281 absl::flat_hash_set<Node*> control_ret_nodes; in Run() local
291 &empty_flib, &control_ret_nodes); in Run()
306 &control_ret_nodes), in Run()
310 control_ret_node_names->reserve(control_ret_nodes.size()); in Run()
311 for (const auto* node : control_ret_nodes) in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_body.cc54 this->control_ret_nodes.reserve(control_ret_node_names.size()); in FunctionBody()
57 this->control_ret_nodes.push_back(n); in FunctionBody()
Dfunction_body.h42 gtl::InlinedVector<Node*, 4> control_ret_nodes; member
Dfunction.cc879 std::unordered_set<StringPiece, StringPieceHasher> control_ret_nodes; in PruneFunctionBody() local
881 control_ret_nodes.insert(control_ret.second); in PruneFunctionBody()
895 (control_ret_nodes.find(n->name()) != control_ret_nodes.end())) { in PruneFunctionBody()
Dinline_function_utils.cc775 for (Node* fbody_node : fbody->control_ret_nodes) { in InlineFunctionBody()
Dprocess_function_library_runtime.cc661 control_ret_node_names->reserve(fbody->control_ret_nodes.size()); in GetGraphAndArgRets()
662 for (const Node* node : fbody->control_ret_nodes) { in GetGraphAndArgRets()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Djson_translate.cc63 absl::flat_hash_set<tensorflow::Node*> control_ret_nodes; in MlirToJSONTranslateFunction() local
66 &control_ret_nodes); in MlirToJSONTranslateFunction()
/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils.cc487 for (const auto& control_ret_node : fbody->control_ret_nodes) { in HashFunction()
551 for (const auto& control_ret_node : this_fbody->control_ret_nodes) { in CheckFunctionsEqualHelper()
555 for (const auto& control_ret_node : that_fbody->control_ret_nodes) { in CheckFunctionsEqualHelper()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc905 control_sources = {fbody.control_ret_nodes.begin(), in ValidateSideEffectsExecution()
906 fbody.control_ret_nodes.end()}; in ValidateSideEffectsExecution()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc815 GetValidControlRets(fbody->control_ret_nodes, *graph); in CompileFunction()