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

Method from_axes

python/tvm/tirx/exec_context.py:131–136  ·  view source on GitHub ↗
(axes: list[tuple[str, AxisRange]])

Source from the content-addressed store, hash-verified

129
130 @staticmethod
131 def from_axes(axes: list[tuple[str, AxisRange]]) -> ActiveSet:
132 shard = [Iter(axis_range.extent, axis_range.stride, name) for name, axis_range in axes]
133 offset = {
134 Axis.get(name): axis_range.offset for name, axis_range in axes if axis_range.offset != 0
135 }
136 return ActiveSet(TileLayout.from_iters(shard, [], offset))
137
138 @property
139 def size(self) -> int:

Callers 2

replace_axisMethod · 0.80
initial_AFunction · 0.80

Calls 4

IterClass · 0.90
ActiveSetClass · 0.85
from_itersMethod · 0.80
getMethod · 0.45

Tested by

no test coverage detected