Searched defs:control_ret_nodes (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | export_graphdef.cc | 410 absl::flat_hash_set<Node*>* control_ret_nodes) { in GetControlRetNodes()
427 absl::flat_hash_set<Node*>* control_ret_nodes) { in Convert()
599 absl::flat_hash_set<Node*> control_ret_nodes; in ConvertLibFunction() local
669 absl::flat_hash_set<Node*>* control_ret_nodes) { in Convert()
724 absl::flat_hash_set<Node*>* control_ret_nodes) { in ConvertMlirToGraph()
735 absl::flat_hash_set<Node*> control_ret_nodes; in ConvertMlirToGraph() local
|
| H A D | import_model.cc | 649 absl::InlinedVector<Node*, 4> control_ret_nodes; in ConvertDeferredFunctions() local
1370 absl::InlinedVector<Node*, 4>* control_ret_nodes) { in GetArgsAndRetsFromFunctionBody()
1551 const absl::InlinedVector<Node*, 4>& control_ret_nodes, in Convert()
1606 const absl::InlinedVector<Node*, 4>& control_ret_nodes) { in ConvertFunctionArgAndRets()
2430 absl::InlinedVector<Node*, 4> control_ret_nodes; in Convert() local
2725 absl::InlinedVector<Node*, 4>* control_ret_nodes) { in GetControlRetsFromGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_graph_to_mlir_pass.cc | 70 absl::flat_hash_set<Node*>& control_ret_nodes, Fprint128 cache_key) { in RunOnGraph()
|
| H A D | dtensor_device.cc | 1258 const absl::flat_hash_set<Node*>& control_ret_nodes, TFE_Context* context, in AddExecutionFunctionDefsToFunctionDefLibrary()
1402 absl::flat_hash_set<Node*> control_ret_nodes; in LowerToSPMDFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/ |
| H A D | json_translate.cc | 63 absl::flat_hash_set<tensorflow::Node*> control_ret_nodes; in MlirToJSONTranslateFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function_body.h | 42 gtl::InlinedVector<Node*, 4> control_ret_nodes; member
|
| H A D | function.cc | 896 std::unordered_set<StringPiece, StringPieceHasher> control_ret_nodes; in PruneFunctionBody() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/ |
| H A D | mlir_graph_optimization_pass.cc | 319 absl::flat_hash_set<Node*> control_ret_nodes; in Run() local
|