Home
last modified time | relevance | path

Searched refs:if_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcond_v2.py93 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/
Dlower_if_op.cc40 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/
DBUILD124 ":if_op",
282 name = "if_op",
283 srcs = ["if_op.cc"],
284 hdrs = ["if_op.h"],
/external/tensorflow/tensorflow/python/kernel_tests/
Dcond_v2_test.py868 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/
Dextract_outside_compilation_pass_test.cc577 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/
Dfunctionalize_control_flow_test.cc121 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()