Searched refs:if_op (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 93 if_op = op.outputs[0].op
94 true_graph, false_graph = _get_func_graphs(if_op)
97 assert true_graph.outer_graph == if_op.graph
98 assert false_graph.outer_graph == if_op.graph
140 if_op._set_func_attr("then_branch", util.create_new_tf_function(true_graph))
141 if_op._set_func_attr("else_branch",
143 if_op._set_type_list_attr("Tout", true_graph.output_types)
144 if_op._set_shape_list_attr("output_shapes", true_graph.output_shapes)
145 if_op._add_outputs(
157 outputs = _build_cond(if_op.inputs[0], true_grad_graph, false_grad_graph,
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 40 CondBuilder(Node* if_op, const string& then_fn_name,
102 CondBuilder::CondBuilder(Node* if_op, const string& then_fn_name, in CondBuilder() argument
105 : if_op_(if_op), in CondBuilder()
108 name_(if_op->name()), in CondBuilder()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 124 ":if_op",
282 name = "if_op",
283 srcs = ["if_op.cc"],
284 hdrs = ["if_op.h"],
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 868 if_op = output.op.inputs[0].op
869 self.assertEqual(if_op.type, "If")
871 self.assertEqual(len(if_op.outputs), 1)
875 self.assertEqual(len(if_op.outputs), 2)
879 self.assertEqual(len(if_op.outputs), 2)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 577 auto if_op = ops::If(s.WithOpName("if"), cond, in TEST_F() local
580 ops::_Retval retval(s.WithOpName("retval"), if_op.output[0], 0); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 121 auto if_op = ops::If(scope.WithOpName(op_name), less, in TEST() local
124 auto id = ops::Identity(scope.WithOpName("cond/Merge"), if_op.output[0]); in TEST()
|