From aca02856dd5ecd8c177bf16993ce2d368ae56d06 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 22 Jul 2019 15:08:27 -0700 Subject: [PATCH] Add missing TfLiteFloat16 specialization to GetTensorData. PiperOrigin-RevId: 259415502 --- tensorflow/lite/kernels/internal/tensor_ctypes.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow/lite/kernels/internal/tensor_ctypes.h b/tensorflow/lite/kernels/internal/tensor_ctypes.h index 8ee95d4d5b3..e2136dc1549 100644 --- a/tensorflow/lite/kernels/internal/tensor_ctypes.h +++ b/tensorflow/lite/kernels/internal/tensor_ctypes.h @@ -28,6 +28,11 @@ inline float* GetTensorData(TfLiteTensor* tensor) { return tensor != nullptr ? tensor->data.f : nullptr; } +template <> +inline TfLiteFloat16* GetTensorData(TfLiteTensor* tensor) { + return tensor != nullptr ? tensor->data.f16 : nullptr; +} + template <> inline uint8_t* GetTensorData(TfLiteTensor* tensor) { return tensor != nullptr ? tensor->data.uint8 : nullptr;