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

Class Ramp

python/tvm/tirx/expr.py:1189–1220  ·  view source on GitHub ↗

Ramp node. Parameters ---------- base : PrimExpr The base expression. stride : PrimExpr The stride of the ramp. lanes : PrimExpr The lanes of the expression. span : Optional[Span] The location of this expression in the source code.

Source from the content-addressed store, hash-verified

1187
1188@tvm_ffi.register_object("tirx.Ramp")
1189class Ramp(PrimExprWithOp):
1190 """Ramp node.
1191
1192 Parameters
1193 ----------
1194 base : PrimExpr
1195 The base expression.
1196
1197 stride : PrimExpr
1198 The stride of the ramp.
1199
1200 lanes : PrimExpr
1201 The lanes of the expression.
1202
1203 span : Optional[Span]
1204 The location of this expression in the source code.
1205 """
1206
1207 base: PrimExpr
1208 stride: PrimExpr
1209 lanes: PrimExpr
1210
1211 def __init__(
1212 self, base: PrimExpr, stride: PrimExpr, lanes: PrimExpr, span: Span | None = None
1213 ) -> None:
1214 self.__init_handle_by_constructor__(
1215 _ffi_api.Ramp,
1216 base,
1217 stride,
1218 lanes,
1219 span, # type: ignore
1220 )
1221
1222
1223@tvm_ffi.register_object("tirx.Broadcast")

Callers 1

__getitem__Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…