| /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 | 3147 def _case_helper(cond_fn, argument
3188 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 | 356 NameAttrList cond_fn, body_fn; in TEST() local
357 cond_fn.set_name("cond"); in TEST()
361 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
390 NameAttrList cond_fn, body_fn; in TEST() local
391 cond_fn.set_name("cond"); in TEST()
395 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
471 NameAttrList cond_fn, fwd_body_fn, bwd_body_fn; in TEST() local
472 cond_fn.set_name("cond"); in TEST()
477 std::initializer_list<Input>{tl}, cond_fn, fwd_body_fn); in TEST()
480 std::initializer_list<Input>{fwd_while_op.output[0]}, cond_fn, in TEST()
|
| D | functionalize_control_flow_test.cc | 262 NameAttrList cond_fn, body_fn; in TEST() local
263 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
271 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
292 InstantiateFunctionForTest(cond_fn.name(), library, &result)); in TEST()
412 NameAttrList cond_fn, body_fn; in TEST() local
413 TF_ASSERT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
421 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
530 NameAttrList cond_fn, body_fn; in TEST() local
531 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
539 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
[all …]
|
| D | xla_compiler_test.cc | 1594 NameAttrList cond_fn, body_fn; in TEST_F() local
1595 cond_fn.set_name("cond"); in TEST_F()
1598 ops::While(scope, std::initializer_list<Input>{arg}, cond_fn, body_fn); in TEST_F()
1676 NameAttrList cond_fn, body_fn; in TEST_F() local
1677 cond_fn.set_name("cond"); in TEST_F()
1680 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 | 631 NameAttrList cond_fn; in TEST_F() local
632 cond_fn.set_name("cond_fn"); in TEST_F()
637 cond_fn, body_fn); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 67 static Status Run(Node* while_op, const NameAttrList& cond_fn, in Run() argument
70 LowerWhileHelper helper(while_op, cond_fn, body_fn, parallel_iterations, in Run()
79 LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn,
179 LowerWhileHelper::LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn, in LowerWhileHelper() argument
189 cond_call_builder_(NewName("cond"), cond_fn.name(), graph->op_registry(), in LowerWhileHelper()
195 for (const auto& i : cond_fn.attr()) { in LowerWhileHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 223 auto cond_fn = module.lookupSymbol<FuncOp>(op.cond()); in LowerWhileOp() local
255 for (Type type : cond_fn.getType().getInputs()) { in LowerWhileOp()
276 Operation* cond_call_op = CallFn(loc, get_cond_arg, cond_fn, &builder); in LowerWhileOp()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 1994 tensorflow::ops::CondGraphBuilderFn cond_fn = in TF_FinishWhileHelper() local
2033 scope, OutputsFromTFOutputs(parent_inputs, num_loop_vars), cond_fn, in TF_FinishWhileHelper()
|