Home
last modified time | relevance | path

Searched refs:input_indices_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/
Ddelegate.cc162 input_indices_.reserve(input_refs.size()); in Prepare()
164 const int64_t object_index = input_indices_.size(); in Prepare()
165 input_indices_.push_back(tensor_index); in Prepare()
187 for (auto index : input_indices_) { in GetRequiredTemporaries()
218 DequantizeInputs(context, input_indices_, quant_conversion_map_)); in Invoke()
231 for (int i = 0; i < input_indices_.size(); ++i) { in SetInputsAndOutputs()
233 i, GetTensorObject(input_indices_[i], context))); in SetInputsAndOutputs()
348 std::vector<int64_t> input_indices_; member in tflite::gpu::__anon84c75a400111::DelegateKernel
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dgpu_api_delegate.cc196 input_indices_.reserve(input_refs.size()); in Prepare()
198 int object_index = input_indices_.size(); in Prepare()
199 input_indices_.push_back(tensor_index); in Prepare()
216 for (auto index : input_indices_) { in SetInputsAndOutputs()
298 std::vector<int64_t> input_indices_; member in tflite::gpu::cl::__anon7ea745220111::Delegate