Supported ops¶
ATen and custom ops registered for PrivateUse1 (device="tvarant").
Unlisted ops fall back to CPU (tensors are copied as needed).
Click an op for signature, parameters, shape, and examples.
Factories & memory¶
| Op | Page |
|---|---|
empty / empty_strided |
empty |
zeros / ones / full |
zeros / ones / full |
clone / contiguous |
clone / contiguous |
Pointwise¶
| Op | Page |
|---|---|
add |
add |
mul |
mul |
sub |
sub |
relu |
relu |
silu |
silu |
neg / abs / exp / sqrt / … |
unary |
Linear algebra & NN¶
| Op | Page |
|---|---|
mm / addmm / matmul / bmm |
mm family |
linear |
linear |
softmax |
softmax |
layer_norm |
layer_norm |
embedding |
embedding |
Custom (torch.ops.tvarant)¶
| Op | Page |
|---|---|
linear_act |
linear_act |
pointwise |
pointwise |