code: def testQuantizedTypeRoundtrip(self):
      with self.session() as session:
          for dtype in self.quantized_tf_types:
              in_values = np.array([1, 2, 3, 4, 5, 6])
              expected = [[1, 2], [3, 4], [5, 6]]
              with self.test_scope():
                  p = array_ops.placeholder(dtype=dtypes.int32)
                  x = math_ops.cast(p, dtype)
                  x = array_ops.reshape(x, [3, 2])
  
              value = session.run(x, {p: in_values})
              self.assertAllEqual(value, expected)
  
file_path: tensorflow/tensorflow/compiler/tests/quantized_ops_test.py
imports: import math
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.compiler.tf2xla.python import xla
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import bitwise_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 1 for calling the QuantizedOpsTest obj's underlying member method code with example usage: obj.testQuantizedTypeRoundtrip() without return types