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

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

View File

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

View File

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

View File

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