class_docstr: Schedule for all the stages.
class_docstr_tok: ["Schedule","for","all","the","stages","."]
code: def __getitem__(self, k):
      if isinstance(k, _tensor.Tensor):
          k = k.op
      if not isinstance(k, _tensor.Operation):
          raise ValueError(
              "Expect schedule key to be Tensor or Operation"
          )
      if k not in self.stage_map:
          raise ValueError(f"Cannot find the operation {k} in schedule")
      return self.stage_map[k]
  
imports: import collections
  import inspect
  from typing import Callable, List
  import tvm._ffi
  from tvm._ffi.base import string_types
  from tvm.ir import container
  from tvm.runtime import Object, convert
  from tvm.tir import Buffer, IndexMap, IterVar, Var
  from .None import _ffi_api
  from .None import tensor
label_desc: Use image node_id 1 for calling the Schedule obj's underlying member method code with example usage: obj.__getitem__(k) and returns: self