class_docstr: TestElementWise
docstr: Create and return the schedule and input args after applying layout transform
docstr_tok: ["Create","and","return","the","schedule","and","input","args","after","applying","layout","transform"]
code: def schedule_args(
      self,
      schedule_type,
      input_shape,
      dtype,
      input_layout,
      output_layout,
      working_layout,
      working_scope,
  ):
      """Create and return the schedule and input args after applying layout transform"""
      if schedule_type == "TE":
          return self._te_schedule_args(
              input_shape,
              dtype,
              input_layout,
              output_layout,
              working_layout,
              working_scope,
          )
      elif schedule_type == "TIR":
          return self._tir_schedule_args(
              input_shape,
              dtype,
              input_layout,
              output_layout,
              working_layout,
              working_scope,
          )
  
      else:
          raise ValueError(f"Unknown schedule type: {schedule_type}")
  
imports: import contextlib
  import numpy
  import pytest
  import tvm
  import tvm.contrib.hexagon
  import tvm.testing
  from tvm import te
  from tvm.contrib.hexagon.pytest_plugin import requires_hexagon_toolchain
  from tvm.tir.stmt_functor import post_order_visit
  from tvm.contrib.hexagon import allocate_hexagon_array
  from .infrastructure import get_hexagon_target
label_desc: Use image node_id 3 for calling the TestElementWise obj's underlying member method code with example usage: obj.schedule_args(schedule_type, input_shape, dtype, input_layout, output_layout, working_layout, working_scope) and returns: self, self