const auto input_element_type_bitwidth =
        input_element_type.getIntOrFloatBitWidth();
    const auto output_element_type_bitwidth =
        output_element_type.getIntOrFloatBitWidth();

    auto is_output_shape_valid_with_small_input_element_type_bitwidth = [&]() {
      if (output_element_type_bitwidth % input_element_type_bitwidth != 0) {
        op.emitOpError() << "output element bitwidth is not multiple "

    const int output_element_type_bitwidth =
        GetTypeBitWidth(output_element_type);

    if (input_element_type_bitwidth < 0 || output_element_type_bitwidth < 0) {
      // Only supports quantized type, int and float types.
      return op.emitOpError("Unsupported element type.");
    }

    if (input_element_type_bitwidth < output_element_type_bitwidth) {