d = _tfr_quant_rescale(y, s, 0)
  e = math_ops.Cast(x=d, DstT=dtypes.int16)
  f = math_ops.Cast(x=e, DstT=dtypes.int8)
  return f


@composite.Composite('TestIdentityNOp')
def _tfr_quant_test_n(x):
  y = _tfr_quant_raw_data(x)
  return y


class TFRGenTestBase(test.TestCase):

  def _check_code(self, tfr_code, exp_tfr_code):