| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_benchmark.py | 54 def cond_fn(x): function
59 cond_fn(0.0)
64 cond_fn(0.0)
78 cond_fn = sess.make_callable(cond_val, [x])
82 cond_fn(0.0)
87 cond_fn(0.0)
|
| D | control_flow_ops.py | 3193 def _case_helper(cond_fn, argument
3234 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 | 248 NodeDef cond_fn; in RunTest() local
249 cond_fn.set_name("cond_node"); in RunTest()
250 cond_fn.set_op("cond_fn"); in RunTest()
251 *(cond_fn.add_input()) = "source"; in RunTest()
253 scope.graph()->AddNode(cond_fn, &status); in RunTest()
388 NameAttrList cond_fn, body_fn; in TEST() local
389 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
397 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
418 InstantiateFunctionForTest(cond_fn.name(), library, &result)); in TEST()
538 NameAttrList cond_fn, body_fn; in TEST() local
[all …]
|
| D | xla_compiler_test.cc | 1681 NameAttrList cond_fn, body_fn; in TEST_F() local
1682 cond_fn.set_name("cond"); in TEST_F()
1685 ops::While(scope, std::initializer_list<Input>{arg}, cond_fn, body_fn); in TEST_F()
1762 NameAttrList cond_fn, body_fn; in TEST_F() local
1763 cond_fn.set_name("cond"); in TEST_F()
1766 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
69 LowerWhileHelper helper(while_op, cond_fn, body_fn, parallel_iterations, in Run()
78 LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn,
180 LowerWhileHelper::LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn, in LowerWhileHelper() argument
192 cond_call_builder_(NewName("cond"), cond_fn.name(), flib_def, in LowerWhileHelper()
198 for (const auto& i : cond_fn.attr()) { in LowerWhileHelper()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | while_loop.py | 58 def cond_fn(counter, value, increment_value): function
74 cond_fn, body_fn, loop_vars=[1, increment_value, increment_value])
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | while_loop.py | 62 def cond_fn(counter, value, increment_value): function
78 cond_fn, body_fn, loop_vars=[1, increment_value, increment_value])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 196 auto cond_fn = op.cond_function(); in LowerWhileOp() local
228 for (Type type : cond_fn.getFunctionType().getInputs()) { in LowerWhileOp()
249 Operation* cond_call_op = CallFn(loc, get_cond_arg, cond_fn, &builder); in LowerWhileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | tf_to_tfrt.cc | 1200 mlir::FlatSymbolRefAttr cond_fn = op.condAttr(); in matchAndRewrite() local
1223 GetPredicateFunction(op, cond_fn, while_arg_result_types, rewriter); in matchAndRewrite()
1282 mlir::TF::WhileOp op, mlir::FlatSymbolRefAttr cond_fn,
1310 mlir::TF::WhileOp op, mlir::FlatSymbolRefAttr cond_fn, in GetPredicateFunction() argument
1313 std::string pred_fn_name = cond_fn.getValue().str() + "/tfrt_predicate"; in GetPredicateFunction()
1344 op.getLoc(), cond_result_types, cond_fn, block->getArguments()); in GetPredicateFunction()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_test.py | 1041 def _fixed_while_loop(self, cond_fn): argument
1043 return cond_fn(s)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3115 auto cond_fn = in verifySymbolUses() local
3119 if (!cond_fn) { in verifySymbolUses()
3126 auto cond_fn_type = cond_fn.getFunctionType(); in verifySymbolUses()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2022 tensorflow::ops::CondGraphBuilderFn cond_fn = in TF_FinishWhileHelper() local
2061 scope, OutputsFromTFOutputs(parent_inputs, num_loop_vars), cond_fn, in TF_FinishWhileHelper()
|