From a74e357be70af48a95238cef74e502815bb3d6e0 Mon Sep 17 00:00:00 2001 From: ddavis-2015 Date: Wed, 26 Jul 2023 09:18:06 -0700 Subject: [PATCH] Add the AddEmbeddingLookup method to micro_mutable_op_resolver.h @tensorflow/micro Update micro_mutable_op_resolver.h bug=fixes AddEmbeddingLookup not in micro_mutable_op_resolver.h #2135 --- tensorflow/lite/micro/micro_mutable_op_resolver.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow/lite/micro/micro_mutable_op_resolver.h b/tensorflow/lite/micro/micro_mutable_op_resolver.h index ae2754b4187..fe4f5fbc139 100644 --- a/tensorflow/lite/micro/micro_mutable_op_resolver.h +++ b/tensorflow/lite/micro/micro_mutable_op_resolver.h @@ -224,6 +224,11 @@ class MicroMutableOpResolver : public MicroOpResolver { 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());