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 | 65 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 | 2442 status->status = tensorflow::ops::BuildWhileLoop( in TF_FinishWhileHelper()
|