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

Method as_struct_info

python/tvm/relax/script/parser/entry.py:435–437  ·  view source on GitHub ↗
(self, dict_globals: dict[str, Any] | None = None)

Source from the content-addressed store, hash-verified

433 return {v for v in self.values if isinstance(v, str) and v.isidentifier()}
434
435 def as_struct_info(self, dict_globals: dict[str, Any] | None = None) -> ShapeStructInfo:
436 values = [_eval_shape(v, dict_globals) for v in self.values] if self.values else None
437 return ShapeStructInfo(values, self.ndim)
438
439
440def Shape(values: list[PrimExpr] | None = None, ndim: int = -1) -> ShapeProxy:

Callers

nothing calls this directly

Calls 2

ShapeStructInfoClass · 0.90
_eval_shapeFunction · 0.85

Tested by

no test coverage detected