From 182c5c6f2a995c6c56c858b1057e4054adc2a94e Mon Sep 17 00:00:00 2001
From: Feng Liu <fengliuai@google.com>
Date: Fri, 5 Mar 2021 17:03:35 -0800
Subject: [PATCH] Mark Conv3D to be not quantizable
PiperOrigin-RevId: 361254049
Change-Id: I8b4ff148912766020f107cb045f5bab52092f71c
---
tensorflow/compiler/mlir/lite/ir/tfl_ops.td | 1 +
1 file changed, 1 insertion(+)
diff --git a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
index 74d2a66cc0b..8c3dd2a365c 100644
--- a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
+++ b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td
@@ -4634,6 +4634,7 @@ Read variable data identified by 'resource_id'.
def TFL_Conv3DOp : TFL_Op<"conv_3d", [
NoSideEffect,
+ NoQuantizableResult,
AccumulatorUniformScale<2, 0, 1>,
TFL_OperandHasRank<0, 5>,
TFL_OperandHasRank<1, 5>,