# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -o - -mlir-print-debuginfo -mlir-print-local-scope | FileCheck %s

node {
  name: "Quantized_Constant"
  op: "Const"
  attr {
    key: "dtype"
    value {
      type: DT_QUINT8
    }
  }
  attr {
    key: "value"
    value {
      tensor {
        dtype: DT_QUINT8
        tensor_shape {
        }
        int_val: 7
      }
    }

  %6 = "tfl.dequantize"(%5) : (tensor<1x401408x!quant.uniform<u8:f32, 3.906250e-03>>) -> tensor<1x401408xf32>
  func.return %6 : tensor<1x401408xf32>
}

// CHECK-LABEL: quantized_constant
func.func @quantized_constant(%arg0: tensor<1x2xf32>) -> tensor<2x2xf32> {
  %1 = "tfl.quantize"(%arg0) {qtype = tensor<1x2x!quant.uniform<u8:f32, 1.0>>, volatile} : (tensor<1x2xf32>) -> tensor<1x2x!quant.uniform<u8:f32, 1.0>>

// CHECK: stablehlo.add
// CHECK-NOT: tfl.add

// -----

// Tests that a quantized `stablehlo.constant` is converted into `tfl.qconst`.

// CHECK-LABEL: func @quantized_constant
func.func @quantized_constant() -> tensor<1x2x4x5x!quant.uniform<i8:f32, 1.000000e+0>> {
  %0 = stablehlo.constant() {value = dense<1> : tensor<1x2x4x5xi8>} : () -> tensor<1x2x4x5x!quant.uniform<i8:f32, 1.000000e+0>>