Home
last modified time | relevance | path

Searched refs:BuildWhileLoop (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcontrol_flow_test.cc43 return ops::BuildWhileLoop(scope.NewSubScope("inner"), inputs, in NestedLoopBody()
53 TF_ASSERT_OK(ops::BuildWhileLoop(scope.NewSubScope("outer"), inputs, in TEST()
82 TF_ASSERT_OK(ops::BuildWhileLoop(scope, inputs, LessThanTenCond, AddOneBody, in TEST()
125 TF_ASSERT_OK(ops::BuildWhileLoop(scope, inputs, LessThanTenCond, AddOneBody, in TEST()
128 TF_ASSERT_OK(ops::BuildWhileLoop(scope.NewSubScope("sub"), inputs, in TEST()
Dgraph_partition_test.cc417 TF_ASSERT_OK(ops::BuildWhileLoop( in TEST_F()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients.cc28 using ops::BuildWhileLoop;
85 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, {zero}, cond_fn, body_fn, in AddForwardLoopCounter()
124 TF_RETURN_IF_ERROR(BuildWhileLoop( in AddBackPropLoopCounter()
171 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, grad_inputs, cond_fn, body_fn, in AddWhileGradientLoop()
Dwhile_gradients_test.cc44 TF_ASSERT_OK(ops::BuildWhileLoop(scope_, *inputs, cond, body, "test_loop", in CreateLoop()
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop.h65 Status BuildWhileLoop(const Scope& scope, const std::vector<Output>& inputs,
Dwhile_loop_test.cc43 ops::BuildWhileLoop(scope_, inputs_, cond, body, kFrameName, &outputs_); in CreateLoop()
Dwhile_loop.cc172 Status BuildWhileLoop(const Scope& scope, const std::vector<Output>& inputs, in BuildWhileLoop() function
/external/tensorflow/tensorflow/c/
Dc_api.cc2442 status->status = tensorflow::ops::BuildWhileLoop( in TF_FinishWhileHelper()