| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 75 BodyGraphBuilderFn body_fn = [](const Scope& scope, in AddForwardLoopCounter() local
114 BodyGraphBuilderFn body_fn = [](const Scope& scope, in AddBackPropLoopCounter() local
160 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope, in AddWhileGradientLoop() local
|
| /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 | 620 NameAttrList body_fn; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 66 const NameAttrList& body_fn, int parallel_iterations, in Run()
178 const NameAttrList& body_fn, in LowerWhileHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 78 auto body_fn = [&](XlaOp i, absl::Span<const XlaOp> loop_vars, in CholeskyUnblocked() local
|
| D | qr_expander.cc | 303 auto body_fn = [&](XlaOp j, absl::Span<const XlaOp> values, in CompactWYRepresentation() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | constraints.py | 256 def body_fn(i, array): function
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_test.py | 549 def _basic_loop(self, init_value, body_fn): argument
848 def _basic_loop(self, init_value, body_fn): argument
1092 def _basic_cond(self, body_fn, else_fn): argument
|
| /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
|
| D | functionalize_control_flow_test.cc | 369 NameAttrList cond_fn, body_fn; in TEST() local
519 NameAttrList cond_fn, body_fn; in TEST() local
637 NameAttrList cond_fn, body_fn; in TEST() local
785 NameAttrList cond_fn, body_fn; in TEST() local
|
| D | xla_compiler_test.cc | 1602 NameAttrList cond_fn, body_fn; in TEST_F() local
1683 NameAttrList cond_fn, body_fn; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 195 auto body_fn = op.body_function(); in LowerWhileOp() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 112 def body_fn(v): # pylint: disable=invalid-name function
323 def body_fn(i): # pylint: disable=invalid-name function
|
| D | control_flow_ops_py_test.py | 1551 def body_fn(i): function
1568 def body_fn(unused_i): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 436 def body_fn(i): function
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 1497 def body_fn(t, state, ta): function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2896 auto body_fn = op.body_function(); in Verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 646 WhileBodyFnType body_fn, ArrayRef<Value> init_values, in CreateWhile32()
6000 auto body_fn = [&](Location loc, Value j, ArrayRef<Value> old_values, in ComputeWYRepresentation() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2013 tensorflow::ops::BodyGraphBuilderFn body_fn = in TF_FinishWhileHelper() local
|