class_docstr: Test batch flatten class.
class_docstr_tok: ["Test","batch","flatten","class","."]
docstr: Test batch flatten.
docstr_tok: ["Test","batch","flatten","."]
code: def test_batch_flatten(
      self,
      data_type,
      input_shape,
      input_layout,
      output_shape,
      output_layout,
      hexagon_session,
  ):
      """Test batch flatten."""
      reshape_helper(
          "batch_flatten",
          sl.batch_flatten_compute,
          sl.batch_flatten_stir_schedule,
          data_type,
          input_shape,
          input_layout,
          output_shape,
          output_layout,
          hexagon_session,
      )
  
file_path: tvm/tests/python/contrib/test_hexagon/topi/test_reshape.py
imports: import numpy
  import tvm
  import tvm.testing
  import tvm.topi.hexagon.slice_ops
  from tvm import te
  from tvm.contrib.hexagon import allocate_hexagon_array
  from ..infrastructure import transform_numpy, get_hexagon_target
label_desc: Use image node_id 1 for calling the TestBatchFlatten obj's underlying member method code with example usage: obj.test_batch_flatten(data_type, input_shape, input_layout, output_shape, output_layout, hexagon_session) without return types