Searched refs:body_outputs (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 154 ASSERT_TRUE(params_->body_outputs != nullptr); in TEST_F()
170 params_->body_outputs[0] = {add2, 0}; in TEST_F()
171 params_->body_outputs[1] = params_->body_inputs[1]; in TEST_F()
258 inner_params.body_outputs[1] = {input2_add, 0}; in TEST_F()
263 inner_params.body_outputs[0] = {inner_input1_add, 0}; in TEST_F()
273 params_->body_outputs[0] = {input1_add, 0}; in TEST_F()
275 params_->body_outputs[1] = inner_outputs[1]; in TEST_F()
298 params_->body_outputs[0] = params_->body_inputs[0]; in TEST_F()
306 params_->body_outputs[0] = params_->body_inputs[0]; in TEST_F()
316 params_->body_outputs[0] = params_->body_inputs[0]; in TEST_F()
[all …]
|
| D | c_api.cc | 1886 params.body_inputs == nullptr || params.body_outputs == nullptr) { in ValidateConstWhileParams()
1900 if (params.body_outputs[i].oper == nullptr) { in ValidateInputWhileParams()
1920 delete[] params->body_outputs; in FreeWhileResources()
1955 TF_Output* body_outputs = new TF_Output[ninputs]; in TF_NewWhile()
1956 for (int i = 0; i < ninputs; ++i) body_outputs[i] = {nullptr, -1}; in TF_NewWhile()
1972 body_graph, body_inputs, body_outputs, name}; in TF_NewWhile()
2023 params->body_outputs, num_loop_vars, outputs)); in TF_FinishWhileHelper()
|
| D | c_api.h | 924 TF_Output* const body_outputs; member
|
| D | c_api_function_test.cc | 934 params->body_outputs[0] = {add2, 0}; in TEST_F()
935 params->body_outputs[1] = params->body_inputs[1]; in TEST_F()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 141 DCHECK_EQ(grad_inputs.size(), while_ctx->body_outputs().size()); in AddWhileGradientLoop()
142 DCHECK_EQ(while_ctx->body_inputs().size(), while_ctx->body_outputs().size()); in AddWhileGradientLoop()
163 std::vector<Output> body_outputs = in AddWhileGradientLoop() local
164 ToOutputVector(while_ctx->body_outputs()); in AddWhileGradientLoop()
166 return AddSymbolicGradients(scope, body_outputs, body_inputs, inputs, in AddWhileGradientLoop()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | while_context.cc | 25 std::vector<OutputTensor> body_outputs) in WhileContext() argument
31 body_outputs_(std::move(body_outputs)) { in WhileContext()
|
| D | while_context.h | 40 std::vector<OutputTensor> body_outputs);
47 const std::vector<OutputTensor>& body_outputs() const { in body_outputs() function
|
| D | graph.cc | 854 std::vector<OutputTensor> body_outputs, in AddWhileContext() argument
860 std::move(body_outputs)))); in AddWhileContext()
|
| D | graph.h | 699 std::vector<OutputTensor> body_outputs,
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | loops.cc | 104 TF_ASSIGN_OR_RETURN(std::vector<XlaOp> body_outputs, in ForEachIndex()
106 updated_values.insert(updated_values.end(), body_outputs.begin(), in ForEachIndex()
107 body_outputs.end()); in ForEachIndex()
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop.cc | 209 std::vector<Output> body_outputs; in BuildWhileLoop() local
210 TF_RETURN_IF_ERROR(CreateBody(scope, body, switch_trues, &body_outputs)); in BuildWhileLoop()
214 next_outputs[i] = NextIteration(scope, body_outputs[i]); in BuildWhileLoop()
238 ToOutputTensors(body_outputs), &while_ctx)); in BuildWhileLoop()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 283 params.body_outputs, params.ninputs, params.ninputs); in Java_org_tensorflow_Graph_whileLoop()
305 params.body_outputs[i] = {body_output_op, in Java_org_tensorflow_Graph_whileLoop()
|