Searched refs:else_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 43 const NameAttrList& else_fn, bool keep_node_fetchable,
114 const NameAttrList& else_fn, bool keep_node_fetchable, in CondBuilder() argument
124 else_call_builder_(NewName("else"), else_fn.name(), graph->op_registry(), in CondBuilder()
134 for (const auto& i : else_fn.attr()) { in CondBuilder()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 47 NameAttrList* then_fn, NameAttrList* else_fn) { in FindIfThenAndElse() argument
55 *else_fn = *result; in FindIfThenAndElse()
144 NameAttrList else_fn; in CheckGraphDef() local
145 TF_EXPECT_OK(FindIfThenAndElse(graph_def, &op_name, &then_fn, &else_fn)); in CheckGraphDef()
148 InstantiateFunctionForTest(else_fn.name(), library, &else_result)); in CheckGraphDef()
159 else_fn, ops::If::OutputShapes({PartialTensorShape()})); in CheckGraphDef()
205 TF_EXPECT_OK(InstantiateFunctionForTest(else_fn.name(), library, &result)); in CheckGraphDef()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_test.py | 1092 def _basic_cond(self, body_fn, else_fn): argument
1099 x = else_fn()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | resource_op_lifting.mlir | 1032 // CHECK-SAME: else_branch = @else_fn, is_stateless = true, then_branch = @then_fn
1039 { else_branch = @else_fn, then_branch = @then_fn, is_stateless = true}
1046 // CHECK-LABEL: func @else_fn
1048 func @else_fn(%arg0: tensor<*x!tf.resource>, %arg1: tensor<*x!tf.resource>) -> tensor<*xi32> {
|