Searched refs:GetTensorObject (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | gpu_api_delegate.cc | 218 runner_->SetInputObject(i++, GetTensorObject(index, context))); in SetInputsAndOutputs()
223 runner_->SetOutputObject(i++, GetTensorObject(index, context))); in SetInputsAndOutputs()
264 TensorObject GetTensorObject(int index, TfLiteContext* context) const { in GetTensorObject() function in tflite::gpu::cl::__anon7ea745220111::Delegate
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | delegate.cc | 233 i, GetTensorObject(input_indices_[i], context))); in SetInputsAndOutputs()
237 i, GetTensorObject(output_indices_[i], context))); in SetInputsAndOutputs()
251 TensorObject GetTensorObject(int index, TfLiteContext* context) const { in GetTensorObject() function in tflite::gpu::__anon84c75a400111::DelegateKernel
|