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

TFLite Python API: allow float64, uint64 and complex128

Browse Source 
I was confused by the error message saying that the interpreter got UNKNOWN type, which actually turns out to be float64.

Also removes a confusing comment. Since the input type is `int` not `enum`. -Wswitch-enum won't help.

PiperOrigin-RevId: 347300840
Change-Id: I732731871e969ae53e88c9cd7ffffdbd1ac8e314
This commit is contained in:
Tiezhen WANG 2020-12-13 18:43:06 -08:00 committed by TensorFlower Gardener
parent edc060801f
commit 1e8c13a7f0
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/lite/python/interpreter_wrapper/numpy.cc
View File

@ -73,6 +73,8 @@ TfLiteType TfLiteTypeFromPyType(int py_type) {
return kTfLiteFloat32; return kTfLiteFloat32;
case NPY_FLOAT16: case NPY_FLOAT16:
return kTfLiteFloat16; return kTfLiteFloat16;
case NPY_FLOAT64:
return kTfLiteFloat64;
case NPY_INT32: case NPY_INT32:
return kTfLiteInt32; return kTfLiteInt32;
case NPY_INT16: case NPY_INT16:
@ -83,6 +85,8 @@ TfLiteType TfLiteTypeFromPyType(int py_type) {
return kTfLiteInt8; return kTfLiteInt8;
case NPY_INT64: case NPY_INT64:
return kTfLiteInt64; return kTfLiteInt64;
case NPY_UINT64:
return kTfLiteUInt64;
case NPY_BOOL: case NPY_BOOL:
return kTfLiteBool; return kTfLiteBool;
case NPY_OBJECT: case NPY_OBJECT:
@ -91,7 +95,8 @@ TfLiteType TfLiteTypeFromPyType(int py_type) {
return kTfLiteString; return kTfLiteString;
case NPY_COMPLEX64: case NPY_COMPLEX64:
return kTfLiteComplex64; return kTfLiteComplex64;
// Avoid default so compiler errors created when new types are made. case NPY_COMPLEX128:
return kTfLiteComplex128;
} }
return kTfLiteNoType; return kTfLiteNoType;
} }