MCPcopy Create free account
hub / github.com/apache/tvm / normalize

Method normalize

python/tvm/relax/block_builder.py:662–675  ·  view source on GitHub ↗

Normalize an Expr to complete its shape and type. Parameters ---------- expr : Expr The input expr. Returns ------- ret : Expr The expr with normalized shape and type.

(self, expr: Expr)

Source from the content-addressed store, hash-verified

660 return self.add_func(func, self._func._name)
661
662 def normalize(self, expr: Expr) -> Expr:
663 """Normalize an Expr to complete its shape and type.
664
665 Parameters
666 ----------
667 expr : Expr
668 The input expr.
669
670 Returns
671 -------
672 ret : Expr
673 The expr with normalized shape and type.
674 """
675 return _ffi_api.BlockBuilderNormalize(self, expr) # type: ignore
676
677 def get(self) -> tvm.IRModule:
678 """Return intermediate IRModule. For the situation where the IRModule is needed in the

Calls

no outgoing calls