| /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 | 3858 def _case_helper(cond_fn, argument
3899 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 | 354 NameAttrList cond_fn, body_fn; in TEST() local
355 cond_fn.set_name("cond"); in TEST()
359 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
388 NameAttrList cond_fn, body_fn; in TEST() local
389 cond_fn.set_name("cond"); in TEST()
393 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
|
| D | functionalize_control_flow_test.cc | 261 NameAttrList cond_fn, body_fn; in TEST() local
262 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
270 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
291 InstantiateFunctionForTest(cond_fn.name(), library, &result)); in TEST()
413 NameAttrList cond_fn, body_fn; in TEST() local
414 TF_ASSERT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
422 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
533 NameAttrList cond_fn, body_fn; in TEST() local
534 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
542 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 732 NameAttrList cond_fn; in TEST_F() local
733 cond_fn.set_name("cond_fn"); in TEST_F()
738 cond_fn, body_fn); in TEST_F()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2404 tensorflow::ops::CondGraphBuilderFn cond_fn = in TF_FinishWhileHelper() local
2443 scope, OutputsFromTFOutputs(parent_inputs, num_loop_vars), cond_fn, in TF_FinishWhileHelper()
|