diff --git a/tensorflow/lite/experimental/delegates/hexagon/builders/tests/mul_test.cc b/tensorflow/lite/experimental/delegates/hexagon/builders/tests/mul_test.cc index 3fd68c30586..db7eef46150 100644 --- a/tensorflow/lite/experimental/delegates/hexagon/builders/tests/mul_test.cc +++ b/tensorflow/lite/experimental/delegates/hexagon/builders/tests/mul_test.cc @@ -26,8 +26,9 @@ class MulOpModel : public SingleOpModelWithHexagon { input1_ = AddInput(input1); input2_ = AddInput(input2); output_ = AddOutput(output); - SetBuiltinOp(BuiltinOperator_MUL, BuiltinOptions_MulOptions, - CreateMulOptions(builder_, activation_type).Union()); + SetBuiltinOp( + BuiltinOperator_MUL, BuiltinOptions_MulOptions, + CreateMulOptions(builder_, ActivationFunctionType_NONE).Union()); BuildInterpreter({GetShape(input1_), GetShape(input2_)}); } diff --git a/tensorflow/lite/experimental/delegates/hexagon/utils.cc b/tensorflow/lite/experimental/delegates/hexagon/utils.cc index b1ecde764ad..508f6657e61 100644 --- a/tensorflow/lite/experimental/delegates/hexagon/utils.cc +++ b/tensorflow/lite/experimental/delegates/hexagon/utils.cc @@ -67,6 +67,8 @@ bool CheckOpVersion(const TfLiteRegistration* registration) { case kTfLiteBuiltinDepthwiseConv2d: case kTfLiteBuiltinSoftmax: return registration->version <= 2; + case kTfLiteBuiltinRelu: + return registration->version >= 2; default: return registration->version == 1; }