code: def batch_flatten_compute(inp: te.Tensor) -> te.Tensor:
      """Compute for slice batch flatten op for hexagon.
      This op makes the following assumptions:
      1. This op is written for a sliced batch flatten operation.
      2. The input is assumed to be in NHWC layout.
  
      Parameters
      ----------
      Input : te.Tensor
          Input activations padded for inner dimension size
      Returns
      -------
      Output : te.Tensor
          Output of applying batch flatten operation on input
      """
      return topi.nn.flatten(inp)
  
imports: from tvm import te, tir, topi
  from ..utils import get_layout_transform_fn
label_desc: Use image node_id 1 for calling a global function with example usage: batch_flatten_compute(inp) and returns: topi