Searched refs:input_tensors_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/testing/ |
| D | tf_driver.cc | 114 input_tensors_.reserve(input_layer.size()); in TfDriver()
119 input_tensors_[input_layer[i]] = {}; in TfDriver()
191 input_tensors_[input_names_[id]] = tensor; in SetInput()
196 auto tensor = input_tensors_[input_names_[id]]; in ResetTensor()
216 input_tensors_[input_names_[id]] = in ReshapeTensor()
250 auto status = session_->Run({input_tensors_.begin(), input_tensors_.end()}, in Invoke()
|
| D | tf_driver.h | 69 std::unordered_map<string, tensorflow::Tensor> input_tensors_; variable
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 106 input_tensors_ = []
115 input_tensors_.append(input_tensor)
120 input_tensors_.append(input_tensor)
121 input_tensors = input_tensors_
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils_test.cc | 93 input_tensors_, /*dry_run_inference*/ true, &graph_def_with_shapetype)); in FuseByInOut()
130 input_tensors_, /*dry_run_inference=*/true, &graph_def_); in BuildAndAddTensorShape()
144 graph_def_, input_tensors_, &graph_def_); in FuseByPlacedArguments()
151 input_tensors_); in IsFuseReady()
164 const std::vector<std::pair<string, Tensor>> input_tensors_{ member in tensorflow::__anon1120faa10111::FuseRemoteGraphMultipleAddOpsTest
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 227 input_tensors_ = input_tensors; in PreInputInit()
279 CHECK_LE(input_tensors_.size(), inputs_.size()); in PostInputInit()
280 input_tensors_.resize(inputs_.size()); in PostInputInit()
1169 } else if (requested_input_tensor_[i] && i < input_tensors_.size() && in AttachContext()
1170 input_tensors_[i] != nullptr) { in AttachContext()
1173 input_tensors_[i]->SummarizeValue(256 /* max_values */), ">")); in AttachContext()
|
| D | shape_inference.h | 299 return input_tensors_[idx]; in input_tensor()
314 input_tensors_ = input_tensors; in set_input_tensors()
751 std::vector<const Tensor*> input_tensors_; variable
|