Home
last modified time | relevance | path

Searched defs:control_node (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dvariable_merger_pass.cc94 Status MergeReadVariableOps(Node* handle_op, Node* control_node, in MergeReadVariableOps()
173 Node* control_node = nullptr; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcontrol_flow_deps_to_chains.cc105 const auto& control_node = body_graph->node_def( in Run() local
237 const auto& control_node = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dmutable_graph_view.cc85 const NodeDef& control_node) { in CanDedupControlWithRegularInput()
95 NodeDef* control_node = graph.GetNode(control_node_name); in CanDedupControlWithRegularInput() local
1055 NodeDef* control_node = GetControllingFaninToAdd( in AddControllingFanin() local
1458 NodeDef* control_node = in UpdateAllRegularFaninsToControlling() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dpartitioned_topological_sort.cc34 if (auto control_node = dyn_cast<mlir::TFL::ControlNodeOp>(*op)) { in IsFlexDelegate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc3854 ControlNodeOp control_node = *this; in verify() local