| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 67 CondGraphBuilderFn cond_fn = [while_ctx](const Scope& scope, in AddForwardLoopCounter() local
105 CondGraphBuilderFn cond_fn = [](const Scope& scope, in AddBackPropLoopCounter() local
151 CondGraphBuilderFn cond_fn = [backprop_execution_pred]( in AddWhileGradientLoop() local
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | while_loop.py | 58 def cond_fn(counter, value, increment_value): function
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | while_loop.py | 62 def cond_fn(counter, value, increment_value): function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_benchmark.py | 54 def cond_fn(x): function
|
| D | control_flow_ops.py | 3193 def _case_helper(cond_fn, argument
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | rearrange_function_argument_pass_test.cc | 101 NameAttrList cond_fn, body_fn; in TEST() local
214 NameAttrList cond_fn, body_fn; in TEST() local
|
| D | extract_outside_compilation_pass_test.cc | 618 NameAttrList cond_fn; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 65 static Status Run(Node* while_op, const NameAttrList& cond_fn, in Run()
180 LowerWhileHelper::LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn, in LowerWhileHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 359 NameAttrList cond_fn, body_fn; in TEST() local
393 NameAttrList cond_fn, body_fn; in TEST() local
474 NameAttrList cond_fn, fwd_body_fn, bwd_body_fn; in TEST() local
|
| D | functionalize_control_flow_test.cc | 248 NodeDef cond_fn; in RunTest() local
388 NameAttrList cond_fn, body_fn; in TEST() local
538 NameAttrList cond_fn, body_fn; in TEST() local
656 NameAttrList cond_fn, body_fn; in TEST() local
804 NameAttrList cond_fn, body_fn; in TEST() local
|
| D | xla_compiler_test.cc | 1681 NameAttrList cond_fn, body_fn; in TEST_F() local
1762 NameAttrList cond_fn, body_fn; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 196 auto cond_fn = op.cond_function(); in LowerWhileOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | tf_to_tfrt.cc | 1200 mlir::FlatSymbolRefAttr cond_fn = op.condAttr(); in matchAndRewrite() local
1310 mlir::TF::WhileOp op, mlir::FlatSymbolRefAttr cond_fn, in GetPredicateFunction()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_test.py | 1041 def _fixed_while_loop(self, cond_fn): argument
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2022 tensorflow::ops::CondGraphBuilderFn cond_fn = in TF_FinishWhileHelper() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3115 auto cond_fn = in verifySymbolUses() local
|