Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc372 inline Delegate* GetGpuDelegate(TfLiteNode* node) { in GetGpuDelegate() function
376 inline Delegate* GetGpuDelegate(TfLiteDelegate* delegate) { in GetGpuDelegate() function
386 auto* gpu_delegate = GetGpuDelegate(params->delegate); in DelegatePrepare()
403 const auto status = GetGpuDelegate(node)->Invoke(context); in DelegatePrepare()
425 auto* gpu_delegate = GetGpuDelegate(delegate); in DelegateCopyFromBufferHandle()
438 auto* gpu_delegate = GetGpuDelegate(delegate); in DelegateCopyToBufferHandle()
477 delete tflite::gpu::gl::GetGpuDelegate(delegate); in TfLiteGpuDelegateDelete()
483 auto* gpu_delegate = tflite::gpu::gl::GetGpuDelegate(delegate); in TfLiteGpuDelegateBindBufferToTensor()