| /external/tensorflow/tensorflow/contrib/py2tf/utils/ |
| H A D | multiple_dispatch.py | 54 def run_while(cond_fn, body_fn, init_args):
59 body_fn: A Python callable implementing the body of the loop.
61 cond_fn and body_fn.
81 return control_flow_ops.while_loop(cond_fn, body_fn, init_args)
83 return py_while_loop(cond_fn, body_fn, init_args)
86 def py_while_loop(cond_fn, body_fn, init_args):
89 state = body_fn(*state)
|
| H A D | multiple_dispatch_test.py | 46 body_fn = lambda x, t, s: (x * s, t, s)
48 x, _, _ = multiple_dispatch.run_while(cond_fn, body_fn, [3.0, 1.0, 0.5])
51 x, _, _ = multiple_dispatch.run_while(cond_fn, body_fn, [3.0, 4.0, 0.5])
56 body_fn = lambda x, t, s: (x * s, t, s)
59 x, _, _ = multiple_dispatch.run_while(cond_fn, body_fn,
63 x, _, _ = multiple_dispatch.run_while(cond_fn, body_fn,
|
| /external/tensorflow/tensorflow/cc/framework/ |
| H A D | while_gradients.cc | 75 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope, local
85 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, {zero}, cond_fn, body_fn,
114 BodyGraphBuilderFn body_fn = [](const Scope& scope, local
125 scope, {loop_count}, cond_fn, body_fn, frame_name, &outputs,
160 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope, local
171 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, grad_inputs, cond_fn, body_fn,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_control_flow_test.cc | 244 NameAttrList cond_fn, body_fn; local
245 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn));
253 std::initializer_list<Input>{source}, cond_fn, body_fn);
294 TF_EXPECT_OK(InstantiateFunctionForTest(body_fn.name(), library, &result));
347 NameAttrList cond_fn, body_fn; local
348 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn));
356 std::initializer_list<Input>{source}, cond_fn, body_fn);
396 TF_EXPECT_OK(InstantiateFunctionForTest(body_fn.name(), library, &result));
488 NameAttrList cond_fn, body_fn; local
489 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn));
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | gather_op.cc | 111 auto body_fn = [&](xla::ComputationDataHandle i, local
158 TF_ASSIGN_OR_RETURN(auto outputs, XlaForEachIndex(num_indices, ptype, body_fn,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| H A D | scatter.cc | 114 auto body_fn = [&](xla::ComputationDataHandle i, local
188 body_fn, init, "scatter", builder));
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | control_flow_ops_py_test.py | 713 def body_fn(i): function in function:ControlFlowTest.testWhileExternalControlDependencies
718 body=body_fn, loop_vars=[1])
728 def body_fn(unused_i): function in function:ControlFlowTest.testWhileExternalControlDependenciesNoInput
733 body=body_fn, loop_vars=[0])
|
| /external/tensorflow/tensorflow/c/ |
| H A D | c_api.cc | 2300 // 'body_fn' copies the body graph into the parent graph.
2301 tensorflow::ops::BodyGraphBuilderFn body_fn = local
2325 body_fn, params->name, &loop_outputs);
|