// CHECK: %[[CONVERT_1:.*]] = mhlo.bitcast_convert %[[QUANTIZE]] : (tensor<2x!quant.uniform<i8:f32, 1.000000e+00:3>>) -> tensor<2xi8>
  // CHECK: %[[CONVERT_2:.*]] = mhlo.bitcast_convert %[[CONVERT_1]] : (tensor<2xi8>) -> tensor<2x!quant.uniform<i8:f32, 1.000000e+00:3>>
  // CHECK: %[[DEQUANTIZE:.*]] = mhlo.uniform_dequantize %[[CONVERT_2]] : (tensor<2x!quant.uniform<i8:f32, 1.000000e+00:3>>) -> tensor<2xf32>

  %5 = "tf.Range"(%4, %3, %2) : (tensor<i32>, tensor<i32>, tensor<i32>) -> tensor<4096xi32>
  %6 = "tf.Equal"(%1, %5) : (tensor<*xi32>, tensor<4096xi32>) -> tensor<*xi1>
  %7 = "tf.Cast"(%6) : (tensor<*xi1>) -> tensor<*xf32>
  %8 = "tf.BatchMatMulV2"(%7, %arg0) {adj_x = false, adj_y = false} : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32>
  func.return %8 : tensor<*xf32>
}

\337!\002\231\340?\320\225E\031\370+\301?\240\231\222J\010O\316?DB\251jJT\306?\336\303\213\314\022\035\350?\246\324\232}b-\341?\300\002F\240<-\334?\270\254\214\227\033\226\311?\323qE}\346\005\353?\034\226,\211k\377\352?\233\261\352\000\004\210\350?2xIR\334\037\352?\376\306\232\237\307j\334?\340[\227\264\273)\240?|~\277\016I\027\303?\240#\t\276\364\020\351?0>\215\225\322k\256?@g\231%\367\336\205?*\322STU\331\336?f%p\362U\246\343?$3}O\223*\312?\344|\271%U\266\350?l!\371\020#`\337?x\315L\242>\036\3...