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 | 50 NameAttrList* then_fn, NameAttrList* else_fn) { in FindIfThenAndElse() argument
58 *else_fn = *result; in FindIfThenAndElse()
151 NameAttrList else_fn; in CheckGraphDef() local
152 TF_EXPECT_OK(FindIfThenAndElse(graph_def, &op_name, &then_fn, &else_fn)); in CheckGraphDef()
155 InstantiateFunctionForTest(else_fn.name(), library, &else_result)); in CheckGraphDef()
166 else_fn, ops::If::OutputShapes({PartialTensorShape()})); in CheckGraphDef()
212 TF_EXPECT_OK(InstantiateFunctionForTest(else_fn.name(), library, &result)); in CheckGraphDef()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_test.py | 1244 def _basic_cond(self, body_fn, else_fn): argument
1251 x = else_fn()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | resource_op_lifting.mlir | 1061 // CHECK-SAME: else_branch = @else_fn, is_stateless = true, then_branch = @then_fn
1068 { else_branch = @else_fn, then_branch = @then_fn, is_stateless = true}
1075 // CHECK-LABEL: func @else_fn
1077 func.func @else_fn(%arg0: tensor<*x!tf_type.resource>, %arg1: tensor<*x!tf_type.resource>) -> tenso…
|