Searched refs:body_outputs (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 155 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 …]
|
| D | c_api.cc | 2174 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()
|
| D | c_api.h | 1079 TF_Output* const body_outputs; member
|
| D | c_api_function_test.cc | 931 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/ |
| 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 | 689 std::vector<OutputTensor> body_outputs, in AddWhileContext() argument
695 std::move(body_outputs)))); in AddWhileContext()
|
| D | graph.h | 583 std::vector<OutputTensor> body_outputs,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | while_loop.cc | 106 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()
|
| /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()
|