Home
last modified time | relevance | path

Searched refs:control_node (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dpin-ops-with-side-effects.mlir16 // CHECK-NEXT: %[[CONTROL:.*]] = tfl.control_node controls "tf.If"
34 // CHECK-NEXT: %[[CONTROL:.*]] = tfl.control_node controls "tfl.if"
56 // CHECK-NEXT: %[[CONTROL:.*]] = tfl.control_node controls "tfl.while"
70 // CHECK-NEXT: %[[OUT:.*]], %[[CONTROL:.*]] = tfl.control_node controls "tfl.add"(%[[CONST]]
78 // CHECK-NEXT: %[[VAR:.*]], %[[CONTROL:.*]] = tfl.control_node controls "tfl.var_handle"
87 // CHECK-NEXT: control_node controls "tfl.call_once"
114 // CHECK-NEXT: %[[VAR:.*]], %[[SEQ0:.*]] = tfl.control_node controls "tfl.var_handle"()
117 // CHECK-NEXT: %[[TMP2:.*]], %[[SEQ1:.*]] = tfl.control_node(%[[SEQ0]]) controls "tfl.add"(%[[VAR]]…
119 // CHECK-NEXT: %[[TMP4:.*]], %[[SEQ2:.*]] = tfl.control_node(%[[SEQ1]]) controls "tfl.while"(%[[TMP…
129 // CHECK-NEXT: %[[RESULT:.*]], %[[SEQ3:.*]] = tfl.control_node(%[[SEQ2]]) controls "tf.If"(%[[TMP5]…
Dops.mlir3040 %0, %c0 = "tfl.control_node"() ({
3051 …%0, %c0 = tfl.control_node() controls "tfl.add"(%arg0, %arg1) {fused_activation_function = "RELU6"…
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc94 Status MergeReadVariableOps(Node* handle_op, Node* control_node, in MergeReadVariableOps() argument
112 if (control_node) graph->AddControlEdge(control_node, node); in MergeReadVariableOps()
173 Node* control_node = nullptr; in Run() local
179 control_node = e->src(); in Run()
185 read_var_ops[std::pair<Node*, Node*>(var_handle_op, control_node)] in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Dcontrol_flow_deps_to_chains.cc105 const auto& control_node = body_graph->node_def( in Run() local
107 const auto& control_attr = control_node.attr(); in Run()
237 const auto& control_node = in Run() local
240 control_node.attr().at("_res_first_used_by").list().s()) { in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpartitioned_topological_sort.cc34 if (auto control_node = dyn_cast<mlir::TFL::ControlNodeOp>(*op)) { in IsFlexDelegate() local
35 return IsFlexDelegate(&control_node.body().front().front()); in IsFlexDelegate()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dsimple_with_unconnected_control_nodes.mlir117 …%1,%control_1 = tfl.control_node controls "tfl.sub" (%arg0, %0) {fused_activation_function = "RELU…
119 …%3,%control_2 = tfl.control_node controls "tfl.add" (%2, %1) {fused_activation_function = "NONE"} …
Dsimple_with_connected_control_nodes.mlir122 …%1,%control_1 = tfl.control_node controls "tfl.sub" (%arg0, %0) {fused_activation_function = "RELU…
124 …%3,%control_2 = tfl.control_node(%control_1) controls "tfl.add" (%2, %1) {fused_activation_functio…
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc85 const NodeDef& control_node) { in CanDedupControlWithRegularInput() argument
86 return !IsIdentityConsumingSwitch(graph, control_node); in CanDedupControlWithRegularInput()
95 NodeDef* control_node = graph.GetNode(control_node_name); in CanDedupControlWithRegularInput() local
96 if (control_node == nullptr) { in CanDedupControlWithRegularInput()
99 return CanDedupControlWithRegularInput(graph, *control_node); in CanDedupControlWithRegularInput()
1055 NodeDef* control_node = GetControllingFaninToAdd( in AddControllingFanin() local
1060 if (control_node == nullptr) { in AddControllingFanin()
1061 control_node = GetOrCreateIdentityConsumingSwitch(fanin_port); in AddControllingFanin()
1063 AddFaninInternal(node, {control_node, Graph::kControlSlot}); in AddControllingFanin()
1458 NodeDef* control_node = in UpdateAllRegularFaninsToControlling() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc3854 ControlNodeOp control_node = *this; in verify() local
3855 if (!control_node.GetBody().args_empty()) in verify()
3856 return control_node.emitOpError() << "expects body without any arguments"; in verify()
3858 Operation &yield = control_node.GetBody().back(); in verify()
3866 control_node.getNumResults() - 1; // 1 for control token in verify()
3873 if (control_node.getResult(operand_idx).getType() != in verify()
Dtfl_ops.td5634 def TFL_ControlNodeOp : Op<TFL_Dialect, "control_node",
5637 The `TFL.control_node` operation wraps single-block operations in order to attach control edges.