Searched refs:body_inputs (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | while_loop_test.cc | 151 ASSERT_TRUE(params_->body_inputs != nullptr); in TEST_F()
152 EXPECT_TRUE(params_->body_inputs[0].oper != nullptr); in TEST_F()
153 EXPECT_TRUE(params_->body_inputs[1].oper != nullptr); in TEST_F()
164 TF_Operation* add1 = Add(params_->body_inputs[0], params_->body_inputs[1], in TEST_F()
172 params_->body_outputs[1] = params_->body_inputs[1]; in TEST_F()
236 TF_Output inner_inputs[] = {params_->body_inputs[0], params_->body_inputs[1]}; in TEST_F()
257 Add(inner_params.body_inputs[1].oper, one, inner_params.body_graph, s_); in TEST_F()
261 TF_Operation* inner_input1_add = Add(inner_params.body_inputs[0].oper, two, in TEST_F()
272 Add(params_->body_inputs[0], inner_outputs[1], params_->body_graph, s_); in TEST_F()
299 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()
2207 delete[] params->body_inputs; in FreeWhileResources()
2242 TF_Output* body_inputs = new TF_Output[ninputs]; in TF_NewWhile()
2254 &body_inputs[i], status)) { in TF_NewWhile()
2260 body_graph, body_inputs, body_outputs, name}; in TF_NewWhile()
2309 &parent->refiner, params->body_inputs, inputs, in TF_FinishWhileHelper()
|
| D | c_api_function_test.cc | 924 TF_Operation* add1 = Add(params->body_inputs[0], params->body_inputs[1], in TEST_F()
932 params->body_outputs[1] = params->body_inputs[1]; in TEST_F()
|
| D | c_api.h | 1078 const TF_Output* const body_inputs; member
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients.cc | 142 DCHECK_EQ(while_ctx->body_inputs().size(), while_ctx->body_outputs().size()); in AddWhileGradientLoop()
165 std::vector<Output> body_inputs = ToOutputVector(while_ctx->body_inputs()); in AddWhileGradientLoop() local
166 return AddSymbolicGradients(scope, body_outputs, body_inputs, inputs, in AddWhileGradientLoop()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | while_context.cc | 24 std::vector<OutputTensor> body_inputs, in WhileContext() argument
30 body_inputs_(std::move(body_inputs)), in WhileContext()
|
| D | while_context.h | 39 std::vector<OutputTensor> body_inputs,
46 const std::vector<OutputTensor>& body_inputs() const { return body_inputs_; } in body_inputs() function
|
| D | graph.cc | 688 std::vector<OutputTensor> body_inputs, in AddWhileContext() argument
694 cond_output, std::move(body_inputs), in AddWhileContext()
|
| D | graph.h | 582 std::vector<OutputTensor> body_inputs,
|