// CHECK: return %[[RESULT]]
}

// Checks the `FuseAddAndFullyConnected` pattern is not applied if there is quantized type.
// CHECK-LABEL: @FuseAddWithFullyConnectedWithQuantizedWeight
func.func @FuseAddWithFullyConnectedWithQuantizedWeight(%arg: tensor<2x512xf32>) -> tensor<2x1024xf32> {

  func.return %0 : tensor<4x1xf16>
}

// -----

// CHECK-LABEL: func @max_qint
// Regression test to ensure we don't crash getting the initial value for
// tf.Max when using quantized integer types.
func.func @max_qint(%arg0: tensor<4x8x!tf_type.qint8>) -> tensor<4x1x!tf_type.qint8> {
  %dimension = "tf.Const"() { value = dense<1> : tensor<1xi64> } : () -> tensor<1xi64>

      Region &region = mhlo_op.getBodyRegion(idx);
      rewriter.inlineRegionBefore(op.getBodyRegion(idx), region, region.end());

      // Update region's entry blocks argument types to handle quantized element
      // types.
      if constexpr (std::is_same<DstOpT, mhlo::WhileOp>::value) {
        TypeConverter::SignatureConversion signature(num_results);
        Block &block = region.front();