| /external/tensorflow/tensorflow/python/ops/ |
| control_flow_ops_benchmark.py | 58 def cond_fn(x): function in function:CondWithManyIntermediatesBenchmark._benchmark_defun
63 cond_fn(0.0)
68 cond_fn(0.0)
82 cond_fn = sess.make_callable(cond_val, [x])
86 cond_fn(0.0)
91 cond_fn(0.0)
|
| control_flow_ops.py | [all...] |
| /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/ |
| tf2xla_util_test.cc | 354 NameAttrList cond_fn, body_fn; local
355 cond_fn.set_name("cond");
359 std::initializer_list<Input>{pred, input}, cond_fn, body_fn);
388 NameAttrList cond_fn, body_fn; local
389 cond_fn.set_name("cond");
393 std::initializer_list<Input>{pred, input}, cond_fn, body_fn);
|
| functionalize_control_flow_test.cc | 261 NameAttrList cond_fn, body_fn; local
262 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn));
270 std::initializer_list<Input>{source}, cond_fn, body_fn);
291 InstantiateFunctionForTest(cond_fn.name(), library, &result));
413 NameAttrList cond_fn, body_fn; local
414 TF_ASSERT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn));
422 std::initializer_list<Input>{source}, cond_fn, body_fn);
533 NameAttrList cond_fn, body_fn; local
534 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn));
542 std::initializer_list<Input>{source}, cond_fn, body_fn)
681 NameAttrList cond_fn, body_fn; local
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| extract_outside_compilation_pass_test.cc | 695 // "while0" (input = "const0", cond = "cond_fn", body = "body_fn")
711 TF_CHECK_OK(GraphToFunctionDef(*g, "cond_fn", cond_fn_fdef));
732 NameAttrList cond_fn; local
733 cond_fn.set_name("cond_fn");
738 cond_fn, body_fn);
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | 2404 tensorflow::ops::CondGraphBuilderFn cond_fn = local
[all...] |
