Add the AddEmbeddingLookup method to micro_mutable_op_resolver.h (#2148) @tensorflow/micro Update micro_mutable_op_resolver.h bug=fixes #2135
diff --git a/tensorflow/lite/micro/micro_mutable_op_resolver.h b/tensorflow/lite/micro/micro_mutable_op_resolver.h index ae2754b..fe4f5fb 100644 --- a/tensorflow/lite/micro/micro_mutable_op_resolver.h +++ b/tensorflow/lite/micro/micro_mutable_op_resolver.h
@@ -224,6 +224,11 @@ return AddBuiltin(BuiltinOperator_DIV, tflite::Register_DIV(), ParseDiv); } + TfLiteStatus AddEmbeddingLookup() { + return AddBuiltin(BuiltinOperator_EMBEDDING_LOOKUP, + Register_EMBEDDING_LOOKUP(), ParseEmbeddingLookup); + } + TfLiteStatus AddEnergy() { // TODO(b/286250473): change back name to "Energy" and remove namespace return AddCustom("SignalEnergy", tflite::tflm_signal::Register_ENERGY());