Home
last modified time | relevance | path

Searched refs:body_outputs (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/c/
Dwhile_loop_test.cc155 ASSERT_TRUE(params_->body_outputs != nullptr); in TEST_F()
171 params_->body_outputs[0] = {add2, 0}; in TEST_F()
172 params_->body_outputs[1] = params_->body_inputs[1]; in TEST_F()
259 inner_params.body_outputs[1] = {input2_add, 0}; in TEST_F()
264 inner_params.body_outputs[0] = {inner_input1_add, 0}; in TEST_F()
274 params_->body_outputs[0] = {input1_add, 0}; in TEST_F()
276 params_->body_outputs[1] = inner_outputs[1]; in TEST_F()
299 params_->body_outputs[0] = params_->body_inputs[0]; in TEST_F()
307 params_->body_outputs[0] = params_->body_inputs[0]; in TEST_F()
317 params_->body_outputs[0] = params_->body_inputs[0]; in TEST_F()
[all …]
Dc_api.cc2174 params.body_inputs == nullptr || params.body_outputs == nullptr) { in ValidateConstWhileParams()
2188 if (params.body_outputs[i].oper == nullptr) { in ValidateInputWhileParams()
2208 delete[] params->body_outputs; in FreeWhileResources()
2243 TF_Output* body_outputs = new TF_Output[ninputs]; in TF_NewWhile()
2244 for (int i = 0; i < ninputs; ++i) body_outputs[i] = {nullptr, -1}; in TF_NewWhile()
2260 body_graph, body_inputs, body_outputs, name}; in TF_NewWhile()
2311 params->body_outputs, num_loop_vars, outputs)); in TF_FinishWhileHelper()
Dc_api.h1079 TF_Output* const body_outputs; member
Dc_api_function_test.cc931 params->body_outputs[0] = {add2, 0}; in TEST_F()
932 params->body_outputs[1] = params->body_inputs[1]; in TEST_F()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients.cc141 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/cc/ops/
Dwhile_loop.cc209 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/core/graph/
Dwhile_context.cc25 std::vector<OutputTensor> body_outputs) in WhileContext() argument
31 body_outputs_(std::move(body_outputs)) { in WhileContext()
Dwhile_context.h40 std::vector<OutputTensor> body_outputs);
47 const std::vector<OutputTensor>& body_outputs() const { in body_outputs() function
Dgraph.cc689 std::vector<OutputTensor> body_outputs, in AddWhileContext() argument
695 std::move(body_outputs)))); in AddWhileContext()
Dgraph.h583 std::vector<OutputTensor> body_outputs,
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dwhile_loop.cc106 TF_ASSIGN_OR_RETURN(std::vector<xla::ComputationDataHandle> body_outputs, in XlaForEachIndex()
108 updated_values.insert(updated_values.end(), body_outputs.begin(), in XlaForEachIndex()
109 body_outputs.end()); in XlaForEachIndex()