| /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/tf2xla/ |
| D | tf2xla_util_test.cc | 354 NameAttrList cond_fn, body_fn; in TEST() local
356 body_fn.set_name("body"); in TEST()
359 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
388 NameAttrList cond_fn, body_fn; in TEST() local
390 body_fn.set_name("body"); in TEST()
393 std::initializer_list<Input>{pred, input}, cond_fn, body_fn); in TEST()
404 TF_ASSERT_OK(GetNodeAttr(while_node->def(), "body", &body_fn)); in TEST()
405 const FunctionDef* rewritten_body_fn = fld.Find(body_fn.name()); in TEST()
|
| D | functionalize_control_flow_test.cc | 261 NameAttrList cond_fn, body_fn; in TEST() local
262 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
270 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
313 InstantiateFunctionForTest(body_fn.name(), library, &result)); in TEST()
413 NameAttrList cond_fn, body_fn; in TEST() local
414 TF_ASSERT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
422 std::initializer_list<Input>{source}, cond_fn, body_fn); in TEST()
451 InstantiateFunctionForTest(body_fn.name(), library, &result)); in TEST()
533 NameAttrList cond_fn, body_fn; in TEST() local
534 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 68 auto body_fn = in CholeskyUnblocked() local
132 ForEachIndex(n, S32, body_fn, {a, l}, "unblocked", builder)); in CholeskyUnblocked()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 259 auto body_fn = [&](XlaOp j, absl::Span<const XlaOp> values, in ComputeWYRepresentation() local
302 ForEachIndex(n - 1, S32, body_fn, {w, y, vs, taus}, "wy", builder)); in ComputeWYRepresentation()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 92 def body_fn(i): # pylint: disable=invalid-name function
96 loop = while_loop_v2(lambda i: i < 1, body_fn, [0])
|
| D | control_flow_ops_py_test.py | 1303 def body_fn(i): function
1308 body=body_fn, loop_vars=[1])
1320 def body_fn(unused_i): function
1325 body=body_fn, loop_vars=[0])
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 734 NameAttrList body_fn; in TEST_F() local
735 body_fn.set_name("body_fn"); in TEST_F()
738 cond_fn, body_fn); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 885 def body_fn(t, state, ta): function
906 _, state, ta = control_flow_ops.while_loop(condition_fn, body_fn, [
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2420 tensorflow::ops::BodyGraphBuilderFn body_fn = in TF_FinishWhileHelper() local
2444 body_fn, params->name, &loop_outputs); in TF_FinishWhileHelper()
|