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

[tflite] Add macro registration for type string to typeToTfLiteType.

Browse Source 
PiperOrigin-RevId: 204210338
This commit is contained in:
Eugene Brevdo 2018-07-11 16:26:38 -07:00 committed by TensorFlower Gardener
parent 72e0e3d383
commit 4d04403a3d
3 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/contrib/lite/BUILD
View File

@ -146,6 +146,7 @@ cc_library(
":memory_planner", ":memory_planner",
":schema_fbs_version", ":schema_fbs_version",
":simple_memory_arena", ":simple_memory_arena",
":string",
":util", ":util",
"//tensorflow/contrib/lite/kernels:eigen_support", "//tensorflow/contrib/lite/kernels:eigen_support",
"//tensorflow/contrib/lite/kernels:gemm_support", "//tensorflow/contrib/lite/kernels:gemm_support",

1
tensorflow/contrib/lite/allocation.h
View File

@ -23,6 +23,7 @@ limitations under the License.
#include "tensorflow/contrib/lite/context.h" #include "tensorflow/contrib/lite/context.h"
#include "tensorflow/contrib/lite/error_reporter.h" #include "tensorflow/contrib/lite/error_reporter.h"
#include "tensorflow/contrib/lite/simple_memory_arena.h" #include "tensorflow/contrib/lite/simple_memory_arena.h"
#include "tensorflow/contrib/lite/string.h"
namespace tflite { namespace tflite {

4
tensorflow/contrib/lite/interpreter.h
View File

@ -63,6 +63,10 @@ template <>
constexpr TfLiteType typeToTfLiteType<std::complex<float>>() { constexpr TfLiteType typeToTfLiteType<std::complex<float>>() {
return kTfLiteComplex64; return kTfLiteComplex64;
} }
template <>
constexpr TfLiteType typeToTfLiteType<string>() {
return kTfLiteString;
}
// Forward declare since NNAPIDelegate uses Interpreter. // Forward declare since NNAPIDelegate uses Interpreter.
class NNAPIDelegate; class NNAPIDelegate;