code: def test2DInput(self):
      with self.session() as sess:
          p = array_ops.placeholder(dtypes.float32)
          with self.test_scope():
              op = math_ops._bucketize(p, boundaries=[0, 3, 8, 11])
          expected_out = [[0, 1, 1, 2, 2], [3, 3, 4, 4, 1]]
          self.assertAllEqual(
              expected_out,
              sess.run(op, {p: [[-5, 0, 2, 3, 5], [8, 10, 11, 12, 0]]}),
          )
  
file_path: tensorflow-upstream/tensorflow/compiler/tests/bucketize_op_test.py
imports: from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the BucketizationOpTest obj's underlying member method code with example usage: obj.test2DInput() without return types