Searched refs:GetTensorObject (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | gpu_api_delegate.cc | 217 runner_->SetInputObject(i++, GetTensorObject(index, context))); in SetInputsAndOutputs()
222 runner_->SetOutputObject(i++, GetTensorObject(index, context))); in SetInputsAndOutputs()
263 TensorObject GetTensorObject(int index, TfLiteContext* context) const { in GetTensorObject() function in tflite::gpu::cl::__anon38a7c2710111::Delegate
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/ |
| H A D | delegate.cc | 253 i, GetTensorObject(input_indices_[i], context))); in SetInputsAndOutputs()
257 i, GetTensorObject(output_indices_[i], context))); in SetInputsAndOutputs()
272 TensorObject GetTensorObject(int index, TfLiteContext* context) const { in GetTensorObject() function in tflite::gpu::__anon34a402ef0111::DelegateKernel
|