code: def broadcast_axis(tensor_A, tensor_B):
      """Find out the indices that will have broadcasting"""
      A_broadcast = []
      B_broadcast = []
  
      for i in range(len(tensor_A.shape)):
          if tensor_A.shape[i] == tensor_B.shape[i]:
              A_broadcast.append(1)
              B_broadcast.append(1)
          elif tensor_A.shape[i] == 1:
              A_broadcast.append(0)
              B_broadcast.append(1)
          elif tensor_B.shape[i] == 1:
              A_broadcast.append(1)
              B_broadcast.append(0)
      return A_broadcast, B_broadcast
  
file_path: tvm/python/tvm/topi/hexagon/qnn/qadd_qsub_qmul.py
imports: from tvm import te
  from tvm import tir
  from ..utils import get_layout_transform_fn, get_fixed_point_value
label_desc: Use image node_id 1 for calling a global function with example usage: broadcast_axis(tensor_A, tensor_B) and returns: A_broadcast,  B_broadcast