| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 75 BodyGraphBuilderFn body_fn = [](const Scope& scope, in AddForwardLoopCounter() local
85 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, {zero}, cond_fn, body_fn, in AddForwardLoopCounter()
114 BodyGraphBuilderFn body_fn = [](const Scope& scope, in AddBackPropLoopCounter() local
125 scope, {loop_count}, cond_fn, body_fn, frame_name, &outputs, in AddBackPropLoopCounter()
160 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope, in AddWhileGradientLoop() local
171 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, grad_inputs, cond_fn, body_fn, in AddWhileGradientLoop()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | rearrange_function_argument_pass_test.cc | 101 NameAttrList cond_fn, body_fn; in TEST() local
103 body_fn.set_name("f2"); in TEST()
106 std::initializer_list<Input>{arg0, arg1}, cond_fn, body_fn); in TEST()
214 NameAttrList cond_fn, body_fn; in TEST() local
216 body_fn.set_name("f2"); in TEST()
219 cond_fn, body_fn); in TEST()
|
| D | extract_outside_compilation_pass_test.cc | 633 NameAttrList body_fn; in TEST_F() local
634 body_fn.set_name("body_fn"); in TEST_F()
637 cond_fn, body_fn); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 68 const NameAttrList& body_fn, int parallel_iterations, in Run() argument
70 LowerWhileHelper helper(while_op, cond_fn, body_fn, parallel_iterations, in Run()
80 const NameAttrList& body_fn, int parallel_iterations,
180 const NameAttrList& body_fn, in LowerWhileHelper() argument
191 body_call_builder_(NewName("body"), body_fn.name(), graph->op_registry(), in LowerWhileHelper()
199 for (const auto& i : body_fn.attr()) { in LowerWhileHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 356 NameAttrList cond_fn, body_fn; in TEST() local
358 body_fn.set_name("body"); in TEST()
361 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
390 NameAttrList cond_fn, body_fn; in TEST() local
392 body_fn.set_name("body"); in TEST()
395 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
406 TF_ASSERT_OK(GetNodeAttr(while_node->def(), "body", &body_fn)); in TEST()
407 const FunctionDef* rewritten_body_fn = fld.Find(body_fn.name()); 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()
314 InstantiateFunctionForTest(body_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()
450 InstantiateFunctionForTest(body_fn.name(), library, &result)); in TEST()
530 NameAttrList cond_fn, body_fn; in TEST() local
531 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
[all …]
|
| D | xla_compiler_test.cc | 1594 NameAttrList cond_fn, body_fn; in TEST_F() local
1596 body_fn.set_name("body"); 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
1678 body_fn.set_name("body"); in TEST_F()
1680 scope, std::initializer_list<Input>{arg0, arg1, arg2}, cond_fn, body_fn); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 224 auto body_fn = module.lookupSymbol<FuncOp>(op.body()); in LowerWhileOp() local
258 for (Type type : body_fn.getType().getInputs()) { in LowerWhileOp()
289 Operation* body_call_op = CallFn(loc, get_body_arg, body_fn, &builder); in LowerWhileOp()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | constraints.py | 237 def body_fn(i, array): function
245 body_fn,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 77 auto body_fn = in CholeskyUnblocked() local
115 ForEachIndex(n, S32, body_fn, {a, l, ConstantR0<bool>(builder, false)}, in CholeskyUnblocked()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 284 auto body_fn = [&](XlaOp j, absl::Span<const XlaOp> values, in ComputeWYRepresentation() local
329 TF_ASSIGN_OR_RETURN(auto values, ForEachIndex(n - 1, S32, body_fn, in ComputeWYRepresentation()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 105 def body_fn(v): # pylint: disable=invalid-name function
118 lambda v: v < 8., body_fn, [x], return_same_structure=False)
176 def body_fn(i): # pylint: disable=invalid-name function
180 loop = while_loop_v2(lambda i: i < 1, body_fn, [0])
|
| D | control_flow_ops_py_test.py | 1494 def body_fn(i): function
1499 body=body_fn, loop_vars=[1])
1511 def body_fn(unused_i): function
1516 body=body_fn, loop_vars=[0])
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 384 def body_fn(i): function
387 return control_flow_ops.while_loop_v2(lambda i: i < 2, body_fn, [0])
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 1053 def body_fn(t, state, ta): function
1074 _, state, ta = control_flow_ops.while_loop(condition_fn, body_fn, [
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2010 tensorflow::ops::BodyGraphBuilderFn body_fn = in TF_FinishWhileHelper() local
2034 body_fn, params->name, &loop_outputs); in TF_FinishWhileHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 309 WhileBodyFnType body_fn, ArrayRef<Value> init_values, in CreateWhile32() argument
370 body_fn(loc, old_values[0], llvm::makeArrayRef(old_values).drop_front(), in CreateWhile32()
|