code: def softmax_compute(in_tensor):
      """
      Compute for slice softmax op for hexagon.
      This op makes the following assumptions:
      1. This op is written for a sliced softmax operation.
      2. The input is assumed to be in NC layout.
      """
      return topi.nn.softmax(in_tensor, axis=1)
  
imports: import typing
  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: softmax_compute(in_tensor) and returns: topi