| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_benchmark.py | 58 def cond_fn(x): function
63 cond_fn(0.0)
68 cond_fn(0.0)
82 cond_fn = sess.make_callable(cond_val, [x])
86 cond_fn(0.0)
91 cond_fn(0.0)
|
| D | control_flow_ops.py | 3211 def _case_helper(cond_fn, argument
3252 cond_fn, predicate, true_fn=action, false_fn=fn, **cond_kwargs)
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 67 CondGraphBuilderFn cond_fn = [while_ctx](const Scope& scope, in AddForwardLoopCounter() local
85 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, {zero}, cond_fn, body_fn, in AddForwardLoopCounter()
105 CondGraphBuilderFn cond_fn = [](const Scope& scope, in AddBackPropLoopCounter() local
125 scope, {loop_count}, cond_fn, body_fn, frame_name, &outputs, in AddBackPropLoopCounter()
151 CondGraphBuilderFn cond_fn = [backprop_execution_pred]( in AddWhileGradientLoop() local
171 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, grad_inputs, cond_fn, body_fn, in AddWhileGradientLoop()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 359 NameAttrList cond_fn, body_fn; in TEST() local
360 cond_fn.set_name("cond"); in TEST()
364 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
393 NameAttrList cond_fn, body_fn; in TEST() local
394 cond_fn.set_name("cond"); in TEST()
398 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
474 NameAttrList cond_fn, fwd_body_fn, bwd_body_fn; in TEST() local
475 cond_fn.set_name("cond"); in TEST()
480 std::initializer_list<Input>{tl}, cond_fn, fwd_body_fn); in TEST()
483 std::initializer_list<Input>{fwd_while_op.output[0]}, cond_fn, in TEST()
|
| D | functionalize_control_flow_test.cc | 229 NodeDef cond_fn; in RunTest() local
230 cond_fn.set_name("cond_node"); in RunTest()
231 cond_fn.set_op("cond_fn"); in RunTest()
232 *(cond_fn.add_input()) = "source"; in RunTest()
234 scope.graph()->AddNode(cond_fn, &status); in RunTest()
369 NameAttrList cond_fn, body_fn; in TEST() local
370 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
378 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
399 InstantiateFunctionForTest(cond_fn.name(), library, &result)); in TEST()
519 NameAttrList cond_fn, body_fn; in TEST() local
[all …]
|
| D | xla_compiler_test.cc | 1602 NameAttrList cond_fn, body_fn; in TEST_F() local
1603 cond_fn.set_name("cond"); in TEST_F()
1606 ops::While(scope, std::initializer_list<Input>{arg}, cond_fn, body_fn); in TEST_F()
1683 NameAttrList cond_fn, body_fn; in TEST_F() local
1684 cond_fn.set_name("cond"); in TEST_F()
1687 scope, std::initializer_list<Input>{arg0, arg1, arg2}, cond_fn, body_fn); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | rearrange_function_argument_pass_test.cc | 101 NameAttrList cond_fn, body_fn; in TEST() local
102 cond_fn.set_name("f3"); in TEST()
106 std::initializer_list<Input>{arg0, arg1}, cond_fn, body_fn); in TEST()
214 NameAttrList cond_fn, body_fn; in TEST() local
215 cond_fn.set_name("f1"); in TEST()
219 cond_fn, body_fn); in TEST()
|
| D | extract_outside_compilation_pass_test.cc | 618 NameAttrList cond_fn; in TEST_F() local
619 cond_fn.set_name("cond_fn"); in TEST_F()
624 cond_fn, body_fn); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 65 static Status Run(Node* while_op, const NameAttrList& cond_fn, in Run() argument
68 LowerWhileHelper helper(while_op, cond_fn, body_fn, parallel_iterations, in Run()
77 LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn,
177 LowerWhileHelper::LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn, in LowerWhileHelper() argument
187 cond_call_builder_(NewName("cond"), cond_fn.name(), graph->op_registry(), in LowerWhileHelper()
193 for (const auto& i : cond_fn.attr()) { in LowerWhileHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 194 auto cond_fn = op.cond_function(); in LowerWhileOp() local
226 for (Type type : cond_fn.getType().getInputs()) { in LowerWhileOp()
247 Operation* cond_call_op = CallFn(loc, get_cond_arg, cond_fn, &builder); in LowerWhileOp()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_test.py | 889 def _fixed_while_loop(self, cond_fn): argument
891 return cond_fn(s)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2895 auto cond_fn = op.cond_function(); in Verify() local
2897 if (!cond_fn) { in Verify()
2906 auto cond_fn_type = cond_fn.getType(); in Verify()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 1997 tensorflow::ops::CondGraphBuilderFn cond_fn = in TF_FinishWhileHelper() local
2036 scope, OutputsFromTFOutputs(parent_inputs, num_loop_vars), cond_fn, in TF_FinishWhileHelper()
|