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

Add missing TfLiteFloat16 specialization to GetTensorData.

Browse Source 
PiperOrigin-RevId: 259415502
This commit is contained in:
A. Unique TensorFlower 2019-07-22 15:08:27 -07:00 committed by TensorFlower Gardener
parent 6b0dba99c3
commit aca02856dd
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/kernels/internal/tensor_ctypes.h
View File

@ -28,6 +28,11 @@ inline float* GetTensorData(TfLiteTensor* tensor) {
return tensor != nullptr ? tensor->data.f : nullptr; return tensor != nullptr ? tensor->data.f : nullptr;
} }
template <>
inline TfLiteFloat16* GetTensorData(TfLiteTensor* tensor) {
return tensor != nullptr ? tensor->data.f16 : nullptr;
}
template <> template <>
inline uint8_t* GetTensorData(TfLiteTensor* tensor) { inline uint8_t* GetTensorData(TfLiteTensor* tensor) {
return tensor != nullptr ? tensor->data.uint8 : nullptr; return tensor != nullptr ? tensor->data.uint8 : nullptr;