}

// CHECK-LABEL: testAddInt16
func.func @testAddInt16(tensor<? x i16>, tensor<? x i16>) -> tensor<? x i16> {
^bb0(%arg0: tensor<? x i16>, %arg1: tensor<? x i16>):
  // CHECK: tfl.add %arg0, %arg1 {fused_activation_function = "RELU6"}
  %0 = tfl.add %arg0, %arg1 {fused_activation_function = "RELU6"} : tensor<? x i16>
  func.return %0#0 : tensor<? x i16>
}

// CHECK-LABEL: testSub

    Results are always sorted in the descending order.
  }];

  let arguments = (ins
    TFL_TensorOf<[F32, I8, I16, I32, I64, UI8, QI8, QUI8]>:$input,
    TFL_TensorOf<[I16, I32]>:$k);

  let results = (outs
    TFL_TensorOf<[F32, I8, I16, I32, I64, UI8, QI8, QUI8]>:$values,
    TFL_TensorOf<[I16, I32]>:$indices);

  let builders = [
    OpBuilder<(ins "Value":$input, "Value":$k),

}

// CHECK-LABEL: @testAddV2WithRef
func.func @testAddV2WithRef(%arg0: tensor<!tf_type.int16ref>, %arg1: tensor<i16>) -> tensor<i16> {
  // CHECK: tf.AddV2
  %0 = "tf.AddV2"(%arg0, %arg1) : (tensor<!tf_type.int16ref>, tensor<i16>) -> tensor<i16>
  func.return %0 : tensor<i16>
}

// CHECK-LABEL: @testRealDivWithRef

    %if16 = "tf.IfRegion"(%arg0) ({"tf.Yield"(%if15) : (!tf_variant) -> ()},
                                  {"tf.Yield"(%if15) : (!tf_variant) -> ()})
        {is_stateless = true} : (tensor<i1>) -> !tf_variant
    %if17 = "tf.IfRegion"(%arg0) ({"tf.Yield"(%if16) : (!tf_variant) -> ()},
                                  {"tf.Yield"(%if16) : (!tf_variant) -> ()})

      return rewriter.notifyMatchFailure(
          loc, "output_rank or broadcast_to shape not supported");

    // Allow lowering when the input's elements type is F32, BFloat16, I32 or
    // I16.
    if (!(mlir::isa<BFloat16Type, Float32Type>(element_type) ||
          element_type.isInteger(32) || element_type.isInteger(16)))
      return rewriter.notifyMatchFailure(loc, "element_type_not_supported");

  return quantized_type != nullptr &&
         quantized_type.getStorageTypeIntegralWidth() == 16 &&
         quantized_type.isSigned();
}

// Return true when the given element_type is I16.
bool IsI16Type(Type element_type) {
  return element_type.isInteger(16) && !element_type.isUnsignedInteger();
}

// Return true when the given element_type is I32.
bool IsI32Type(Type element_type) {