class_docstr: A state in the search process. It consists of the current loop structure
  and a list of transformation steps used to construct it.
  Each State corresponds to a specific schedule for its ComputeDAG.
  Parameters
  ----------
  state_object : StateObject
      The StateObject corresponding to C++ internal State object.
  dag : ComputeDAG
      The original ComputeDAG of this State.
  Notes
  -----
  This is a wrapper class of StateObject to deal with copy-on-write property
class_docstr_tok: ["A","state","in","the","search","process",".","It","consists","of","the","current","loop","structure","and","a","list","of","transformation","steps","used","to","construct","it",".","Each","State","corresponds","to","a","specific","schedule","for","its","ComputeDAG",".","Parameters","--","--","--","--","--","state_object",":","StateObject","The","StateObject","corresponding","to","C++","internal","State","object",".","dag",":","ComputeDAG","The","original","ComputeDAG","of","this","State",".","Notes","--","--","-","This","is","a","wrapper","class","of","StateObject","to","deal","with","copy-on-write","property"]
code: def _resolve_stage_id(self, stage_id):
      if isinstance(stage_id, Operation):
          return self.stage_id_map[stage_id]
      if isinstance(stage_id, Tensor):
          return self.stage_id_map[stage_id.op]
      if isinstance(stage_id, int):
          return stage_id
      raise ValueError(
          "Invalid stage: "
          + stage_id
          + " . Expect to be a int, Operation or Tensor"
      )
  
imports: import tvm._ffi
  from tvm.te.tensor import Operation, Tensor
  from tvm.runtime import Object
  from .None import _ffi_api
label_desc: Use image node_id 23 for calling the State obj's underlying member method code with example usage: obj._resolve_stage_id(stage_id) and returns: self, self, stage_id