experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TFLGpuDelegateSetCommandEncoder replaced by TFLGpuDelegateSetCommandBuffer.

Browse Source 
PiperOrigin-RevId: 346436548
Change-Id: I8514531bfcafc21f44c9f0c834cbb57dd73429a3
This commit is contained in:
Raman Sarokin 2020-12-08 16:33:38 -08:00 committed by TensorFlower Gardener
parent 20d69d0834
commit 45d19c8f45
2 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/lite/delegates/gpu/metal_delegate.mm
View File

@ -711,14 +711,6 @@ bool TFLGpuDelegateBindMetalBufferToTensor(TfLiteDelegate* delegate, int tensor_
// Note: This function is not exposed in `metal_delegate.h`, but it's exposed in // Note: This function is not exposed in `metal_delegate.h`, but it's exposed in
// `metal_delegate_internal.h`. // `metal_delegate_internal.h`.
bool TFLGpuDelegateSetCommandEncoder(
TfLiteDelegate* delegate, id<MTLComputeCommandEncoder> encoder) {
auto* metal_delegate = ::tflite::gpu::metal::GetMetalDelegate(delegate);
if (!metal_delegate) return false;
metal_delegate->SetCommandEncoder(encoder);
return true;
}
bool TFLGpuDelegateSetCommandBuffer(TfLiteDelegate* delegate, bool TFLGpuDelegateSetCommandBuffer(TfLiteDelegate* delegate,
id<MTLCommandBuffer> command_buffer) { id<MTLCommandBuffer> command_buffer) {
auto* metal_delegate = ::tflite::gpu::metal::GetMetalDelegate(delegate); auto* metal_delegate = ::tflite::gpu::metal::GetMetalDelegate(delegate);

5
tensorflow/lite/delegates/gpu/metal_delegate_internal.h
View File

@ -33,11 +33,6 @@ bool TFLGpuDelegateBindMetalBufferToTensor(TfLiteDelegate* delegate,
int tensor_index, int tensor_index,
id<MTLBuffer> metal_buffer); id<MTLBuffer> metal_buffer);
// Binds user-defined MTLComputeCommandEncoder. The delegate puts all GPU tasks
// into this encoder instead of the internal encoder.
bool TFLGpuDelegateSetCommandEncoder(TfLiteDelegate* delegate,
id<MTLComputeCommandEncoder> encoder);
// Binds user-defined MTLCommandBuffer. The delegate puts all GPU tasks // Binds user-defined MTLCommandBuffer. The delegate puts all GPU tasks
// into this buffer instead of the internal command buffer. // into this buffer instead of the internal command buffer.
bool TFLGpuDelegateSetCommandBuffer(TfLiteDelegate* delegate, bool TFLGpuDelegateSetCommandBuffer(TfLiteDelegate* delegate,