| /external/tensorflow/tensorflow/cc/framework/ |
| while_gradients.cc | 75 BodyGraphBuilderFn body_fn = [](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/ |
| tf2xla_util_test.cc | 354 NameAttrList cond_fn, body_fn; local
356 body_fn.set_name("body");
359 std::initializer_list<Input>{pred, input}, cond_fn, body_fn);
388 NameAttrList cond_fn, body_fn; local
390 body_fn.set_name("body");
393 std::initializer_list<Input>{pred, input}, cond_fn, body_fn);
404 TF_ASSERT_OK(GetNodeAttr(while_node->def(), "body", &body_fn));
405 const FunctionDef* rewritten_body_fn = fld.Find(body_fn.name());
|
| 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);
313 InstantiateFunctionForTest(body_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);
451 InstantiateFunctionForTest(body_fn.name(), library, &result));
533 NameAttrList cond_fn, body_fn; local
534 TF_EXPECT_OK(FindWhileCondAndBody(graph_def, &cond_fn, &body_fn));
681 NameAttrList cond_fn, body_fn; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| cholesky_expander.cc | 68 auto body_fn = local
132 ForEachIndex(n, S32, body_fn, {a, l}, "unblocked", builder));
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| qr.cc | 259 auto body_fn = [&](XlaOp j, absl::Span<const XlaOp> values, local
302 ForEachIndex(n - 1, S32, body_fn, {w, y, vs, taus}, "wy", builder));
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| extract_outside_compilation_pass_test.cc | 695 // "while0" (input = "const0", cond = "cond_fn", body = "body_fn")
727 TF_CHECK_OK(GraphToFunctionDef(*g, "body_fn", body_fn_fdef));
734 NameAttrList body_fn; local
735 body_fn.set_name("body_fn");
738 cond_fn, body_fn);
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| while_v2_test.py | 92 def body_fn(i): # pylint: disable=invalid-name function in function:WhileV2Test.testExternalControlDependencies
96 loop = while_loop_v2(lambda i: i < 1, body_fn, [0])
|
| control_flow_ops_py_test.py | 1303 def body_fn(i): function in function:ControlFlowTest.testWhileExternalControlDependencies
1320 def body_fn(unused_i): function in function:ControlFlowTest.testWhileExternalControlDependenciesNoInput
[all...] |
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| control_flow_ops_test.py | 885 def body_fn(t, state, ta): function in function:create_dynamic_lstm.loop_fn
906 _, state, ta = control_flow_ops.while_loop(condition_fn, body_fn, [
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | 2420 tensorflow::ops::BodyGraphBuilderFn body_fn = local
[all...] |
