Update op resolvers.

Author: ddavis-2015

python/tflite_micro/python_ops_resolver.cc

@@ -99,6 +99,7 @@ PythonOpsResolver::PythonOpsResolver() {
   AddQuantize();
   AddReadVariable();
   AddReduceMax();
+  AddReduceMin();
   AddRelu();
   AddRelu6();
   AddReshape();

tensorflow/lite/micro/kernels/micro_ops.h

@@ -100,6 +100,7 @@ TFLMRegistration Register_PRELU();
 TFLMRegistration Register_QUANTIZE();
 TFLMRegistration Register_READ_VARIABLE();
 TFLMRegistration Register_REDUCE_MAX();
+TFLMRegistration Register_REDUCE_MIN();
 TFLMRegistration Register_RELU();
 TFLMRegistration Register_RELU6();
 TFLMRegistration Register_RESHAPE();

tensorflow/lite/micro/kernels/reduce_common.cc

@@ -118,9 +118,9 @@ TfLiteStatus EvalIntegerMean(TfLiteContext* context, TfLiteNode* node,
 enum MinMaxEvalType { kEvalMin, kEvalMax };
 
 template <typename T>
-struct MinMaxReducerParams {
-  MinMaxReducerParams() = delete;
-  MinMaxReducerParams(MinMaxEvalType evalType) : type_(evalType){};
+struct MinMaxReducerCompare {
+  MinMaxReducerCompare() = delete;
+  MinMaxReducerCompare(MinMaxEvalType evalType) : type_(evalType) {};
 
   constexpr T initialValue() const {
     return (type_ == kEvalMin) ? std::numeric_limits<T>::max()
@@ -161,7 +161,7 @@ TfLiteStatus EvalMinMaxHelper(TfLiteContext* context, TfLiteNode* node,
       context->GetScratchBuffer(context, op_data->resolved_axis_idx));
   switch (input->type) {
     case kTfLiteFloat32: {
-      MinMaxReducerParams<float> reducer(evalType);
+      MinMaxReducerCompare<float> reducer(evalType);
       TF_LITE_ENSURE(
           context,
           reference_ops::ReduceGeneric<float>(
@@ -173,7 +173,7 @@ TfLiteStatus EvalMinMaxHelper(TfLiteContext* context, TfLiteNode* node,
               reducer.initialValue(), reducer.compare()));
     } break;
     case kTfLiteInt8: {
-      MinMaxReducerParams<int8_t> reducer(evalType);
+      MinMaxReducerCompare<int8_t> reducer(evalType);
       TF_LITE_ENSURE_EQ(context, static_cast<double>(op_data->input_scale),
                         static_cast<double>(op_data->output_scale));
       TF_LITE_ENSURE_EQ(context, op_data->input_zp, op_data->output_zp);

tensorflow/lite/micro/micro_mutable_op_resolver.h

@@ -500,6 +500,11 @@ class MicroMutableOpResolver : public MicroOpResolver {
                       ParseReducer);
   }
 
+  TfLiteStatus AddReduceMin() {
+    return AddBuiltin(BuiltinOperator_REDUCE_MIN, Register_REDUCE_MIN(),
+                      ParseReducer);
+  }
+
   TfLiteStatus AddRelu() {
     return AddBuiltin(BuiltinOperator_RELU, tflite::Register_RELU(), ParseRelu);
   }

tensorflow/lite/micro/tools/benchmarking/op_resolver.h

@@ -104,6 +104,7 @@ inline TfLiteStatus CreateOpResolver(TflmOpResolver& op_resolver) {
   TF_LITE_ENSURE_STATUS(op_resolver.AddQuantize());
   TF_LITE_ENSURE_STATUS(op_resolver.AddReadVariable());
   TF_LITE_ENSURE_STATUS(op_resolver.AddReduceMax());
+  TF_LITE_ENSURE_STATUS(op_resolver.AddReduceMin());
   TF_LITE_ENSURE_STATUS(op_resolver.AddRelu());
   TF_LITE_ENSURE_STATUS(op_resolver.AddRelu6());
   TF_LITE_ENSURE_STATUS(op_resolver.AddReshape());

third_party/xtensa/examples/pytorch_to_tflite/pytorch_op_resolver.cc

@@ -83,6 +83,7 @@ TfLiteStatus InitPytorchOpsResolver(PytorchOpsResolver& resolver) {
   TF_LITE_ENSURE_STATUS(resolver.AddQuantize());
   TF_LITE_ENSURE_STATUS(resolver.AddReadVariable());
   TF_LITE_ENSURE_STATUS(resolver.AddReduceMax());
+  TF_LITE_ENSURE_STATUS(resolver.AddReduceMin());
   TF_LITE_ENSURE_STATUS(resolver.AddRelu());
   TF_LITE_ENSURE_STATUS(resolver.AddRelu6());
   TF_LITE_ENSURE_STATUS(resolver.AddReshape());