Searched refs:BuildWhileLoop (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | control_flow_test.cc | 43 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()
|
| D | graph_partition_test.cc | 417 TF_ASSERT_OK(ops::BuildWhileLoop( in TEST_F()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 28 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()
|
| D | while_gradients_test.cc | 44 TF_ASSERT_OK(ops::BuildWhileLoop(scope_, *inputs, cond, body, "test_loop", in CreateLoop()
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop.h | 68 Status BuildWhileLoop(const Scope& scope, const std::vector<Output>& inputs,
|
| D | while_loop_test.cc | 43 ops::BuildWhileLoop(scope_, inputs_, cond, body, kFrameName, &outputs_); in CreateLoop()
|
| D | while_loop.cc | 172 Status BuildWhileLoop(const Scope& scope, const std::vector<Output>& inputs, in BuildWhileLoop() function
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2035 status->status = tensorflow::ops::BuildWhileLoop( in TF_FinishWhileHelper()
|