Searched refs:then_fn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 42 CondBuilder(Node* if_op, const NameAttrList& then_fn,
113 CondBuilder::CondBuilder(Node* if_op, const NameAttrList& then_fn, in CondBuilder() argument
122 then_call_builder_(NewName("then"), then_fn.name(), graph->op_registry(), in CondBuilder()
129 for (const auto& i : then_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
53 *then_fn = *result; in FindIfThenAndElse()
143 NameAttrList then_fn; in CheckGraphDef() local
145 TF_EXPECT_OK(FindIfThenAndElse(graph_def, &op_name, &then_fn, &else_fn)); in CheckGraphDef()
158 std::initializer_list<Input>{less, y, x}, {DT_INT32}, then_fn, in CheckGraphDef()
182 TF_EXPECT_OK(InstantiateFunctionForTest(then_fn.name(), library, &result)); in CheckGraphDef()
|
| /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}
1055 // CHECK-LABEL: func @then_fn
1057 func @then_fn(%arg0: tensor<*x!tf.resource>, %arg1: tensor<*x!tf.resource>) -> tensor<*xi32> {
|