| /external/tensorflow/tensorflow/contrib/py2tf/utils/ |
| multiple_dispatch.py | 54 def run_while(cond_fn, body_fn, init_args):
58 cond_fn: A Python callable implementing the stop conditions of the loop.
61 cond_fn and body_fn.
65 of the init_args, or any variables closed-over in cond_fn are Tensors,
75 # TODO(alexbw): statically determine all active variables in cond_fn,
78 [c.cell_contents for c in six.get_function_closure(cond_fn) or []])
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):
88 while cond_fn(*state)
[all...] |
| multiple_dispatch_test.py | 45 cond_fn = lambda x, t, s: x > t
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])
55 cond_fn = lambda x, t, s: x > t
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/ |
| while_gradients.cc | 67 CondGraphBuilderFn cond_fn = [while_ctx](const Scope& scope, local
85 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, {zero}, cond_fn, body_fn,
105 CondGraphBuilderFn cond_fn = [](const Scope& scope, local
125 scope, {loop_count}, cond_fn, body_fn, frame_name, &outputs,
151 CondGraphBuilderFn cond_fn = [backprop_execution_pred]( local
171 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, grad_inputs, cond_fn, body_fn,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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);
273 TF_EXPECT_OK(InstantiateFunctionForTest(cond_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);
375 TF_EXPECT_OK(InstantiateFunctionForTest(cond_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/c/ |
| c_api.cc | 2285 tensorflow::ops::CondGraphBuilderFn cond_fn = local
[all...] |
