Searched refs:gpu_delegate (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | gpu_api_delegate.cc | 318 auto* gpu_delegate = GetDelegate(params->delegate); in DelegatePrepare() local
321 const auto status = gpu_delegate->Prepare(context, params); in DelegatePrepare()
327 return gpu_delegate; in DelegatePrepare()
373 auto* gpu_delegate = new tflite::gpu::cl::Delegate(options); in TfLiteGpuDelegateCreate_New() local
374 return gpu_delegate ? gpu_delegate->tflite_delegate() : nullptr; in TfLiteGpuDelegateCreate_New()
384 auto* gpu_delegate = tflite::gpu::cl::GetDelegate(delegate); in TfLiteGpuDelegateBindGlBufferToTensor() local
385 if (!gpu_delegate) { in TfLiteGpuDelegateBindGlBufferToTensor()
388 if (!gpu_delegate->SupportsGlObjects()) { in TfLiteGpuDelegateBindGlBufferToTensor()
399 gpu_delegate->BindGlBufferToTensor(buffer_id, tensor_index, type, layout); in TfLiteGpuDelegateBindGlBufferToTensor()
407 auto* gpu_delegate = tflite::gpu::cl::GetDelegate(delegate); in TfLiteGpuDelegateGetSerializedBinaryCache() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | gl_delegate.cc | 387 auto* gpu_delegate = GetGpuDelegate(params->delegate); in DelegatePrepare() local
390 const auto status = gpu_delegate->Prepare(context, params); in DelegatePrepare()
391 if (status.ok()) return gpu_delegate; in DelegatePrepare()
426 auto* gpu_delegate = GetGpuDelegate(delegate); in DelegateCopyFromBufferHandle() local
427 if (!gpu_delegate) return kTfLiteError; in DelegateCopyFromBufferHandle()
428 const auto status = gpu_delegate->CopyFromBufferHandle(buffer_handle, tensor); in DelegateCopyFromBufferHandle()
439 auto* gpu_delegate = GetGpuDelegate(delegate); in DelegateCopyToBufferHandle() local
440 if (!gpu_delegate) return kTfLiteError; in DelegateCopyToBufferHandle()
441 const auto status = gpu_delegate->CopyToBufferHandle(buffer_handle, tensor); in DelegateCopyToBufferHandle()
473 auto* gpu_delegate = new tflite::gpu::gl::Delegate(options); in TfLiteGpuDelegateCreate() local
[all …]
|
| D | delegate.cc | 372 auto* gpu_delegate = GetDelegate(params->delegate); in DelegatePrepare() local
376 absl::make_unique<DelegateKernel>(gpu_delegate); in DelegatePrepare()
426 auto* gpu_delegate = GetDelegate(delegate); in DelegatePrepare() local
428 GetOpsToReplace(context, gpu_delegate->IsQuantOpsAllowed(), in DelegatePrepare()
429 gpu_delegate->MaxDelegatedPartitions()); in DelegatePrepare()
433 gpu_delegate->num_delegate_kernels()); in DelegatePrepare()
459 auto* gpu_delegate = new tflite::gpu::Delegate(options); in TfLiteGpuDelegateV2Create() local
462 return gpu_delegate ? gpu_delegate->tflite_delegate() : nullptr; in TfLiteGpuDelegateV2Create()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | delegate_testing.cc | 137 auto* gpu_delegate = TfLiteGpuDelegateV2Create(&options); in main() local
138 status = gpu_inference->ModifyGraphWithDelegate(gpu_delegate); in main()
166 TfLiteGpuDelegateV2Delete(gpu_delegate); in main()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/java/src/main/java/org/tensorflow/lite/gpu/ |
| D | BUILD | 6 name = "gpu_delegate",
|
| /external/tensorflow/tensorflow/lite/java/ |
| D | BUILD | 105 …srcs = ["//tensorflow/lite/delegates/gpu/java/src/main/java/org/tensorflow/lite/gpu:gpu_delegate"],
|