Index

_

__add__() (tripy.Tensor method) __call__() (tripy.BatchNorm method) (tripy.Conv method) (tripy.ConvTranspose method) (tripy.Embedding method) (tripy.Executable method) (tripy.GroupNorm method) (tripy.LayerNorm method) (tripy.Linear method) __eq__() (tripy.Tensor method) __floordiv__() (tripy.Tensor method) __ge__() (tripy.Tensor method) __getitem__() (tripy.Tensor method) __gt__() (tripy.Tensor method) __le__() (tripy.Tensor method)

__lt__() (tripy.Tensor method) __matmul__() (tripy.Tensor method) __mod__() (tripy.Tensor method) __mul__() (tripy.Tensor method) __ne__() (tripy.Tensor method) __pow__() (tripy.Tensor method) __radd__() (tripy.Tensor method) __rfloordiv__() (tripy.Tensor method) __rmod__() (tripy.Tensor method) __rmul__() (tripy.Tensor method) __rpow__() (tripy.Tensor method) __rsub__() (tripy.Tensor method) __rtruediv__() (tripy.Tensor method) __sub__() (tripy.Tensor method) __truediv__() (tripy.Tensor method)

A

abs() (in module tripy) all() (in module tripy) allclose() (in module tripy) any() (in module tripy)

arange() (in module tripy) argmax() (in module tripy) argmin() (in module tripy) avgpool() (in module tripy)

B

BatchNorm (class in tripy) bfloat16 (class in tripy) bias (tripy.BatchNorm attribute) (tripy.Conv attribute) (tripy.ConvTranspose attribute) (tripy.GroupNorm attribute) (tripy.LayerNorm attribute) (tripy.Linear attribute)

bool (class in tripy)

C

cast() (in module tripy) compile() (in module tripy) concatenate() (in module tripy) Conv (class in tripy)

ConvTranspose (class in tripy) copy() (in module tripy) cos() (in module tripy) cumsum() (in module tripy)

D

default_stream() (in module tripy) dequantize() (in module tripy) device (class in tripy) dilation (tripy.Conv attribute) (tripy.ConvTranspose attribute) DimensionSize (class in tripy) dtype (class in tripy) (tripy.BatchNorm attribute) (tripy.Conv attribute) (tripy.ConvTranspose attribute) (tripy.Embedding attribute) (tripy.GroupNorm attribute) (tripy.LayerNorm attribute) (tripy.Linear attribute)

E

Embedding (class in tripy) enable_dtype_checking (in module tripy.config) eps (tripy.BatchNorm attribute) (tripy.GroupNorm attribute) (tripy.LayerNorm attribute)

Executable (class in tripy) exp() (in module tripy) expand() (in module tripy)

F

flatten() (in module tripy) flip() (in module tripy) float16 (class in tripy) float32 (class in tripy)

float8 (class in tripy) floating (class in tripy) full() (in module tripy) full_like() (in module tripy)

G

gather() (in module tripy) gelu() (in module tripy)

GroupNorm (class in tripy) groups (tripy.Conv attribute) (tripy.ConvTranspose attribute)

I

input_scale (tripy.Linear attribute) InputInfo (class in tripy) int32 (class in tripy) int4 (class in tripy) int64 (class in tripy)

int8 (class in tripy) integer (class in tripy) iota() (in module tripy) iota_like() (in module tripy) itemsize (tripy.dtype attribute)

L

LayerNorm (class in tripy) Linear (class in tripy) load() (tripy.Executable static method)

load_state_dict() (tripy.Module method) log() (in module tripy) logger (in module tripy)

M

masked_fill() (in module tripy) max() (in module tripy) maximum() (in module tripy) maxpool() (in module tripy) mean() (in module tripy)

minimum() (in module tripy) module tripy.config tripy.types Module (class in tripy)

N

name (tripy.dtype attribute) named_children() (tripy.Module method) named_parameters() (tripy.Module method)

normalized_shape (tripy.LayerNorm attribute) num_channels (tripy.GroupNorm attribute) num_features (tripy.BatchNorm attribute) num_groups (tripy.GroupNorm attribute)

O

ones() (in module tripy)

ones_like() (in module tripy) outer() (in module tripy)

P

pad() (in module tripy) padding (tripy.Conv attribute) (tripy.ConvTranspose attribute)

Parameter (class in tripy) permute() (in module tripy) plugin() (in module tripy) prod() (in module tripy)

Q

quant_dtype (tripy.Linear attribute)

quantize() (in module tripy)

R

relu() (in module tripy) repeat() (in module tripy) reshape() (in module tripy)

resize() (in module tripy) rsqrt() (in module tripy) running_mean (tripy.BatchNorm attribute) running_var (tripy.BatchNorm attribute)

S

save() (tripy.Executable method) shape (tripy.Tensor property) ShapeLike (in module tripy.types) sigmoid() (in module tripy) silu() (in module tripy) sin() (in module tripy) softmax() (in module tripy) split() (in module tripy)

sqrt() (in module tripy) squeeze() (in module tripy) stack() (in module tripy) state_dict() (tripy.Module method) Stream (class in tripy) stride (tripy.Conv attribute) (tripy.ConvTranspose attribute) sum() (in module tripy) synchronize() (tripy.Stream method)

T

tanh() (in module tripy) Tensor (class in tripy) TensorLike (in module tripy.types) timing_cache_file_path (in module tripy.config) transpose() (in module tripy) tril() (in module tripy)

tripy.config module tripy.types module TripyException (class in tripy) triu() (in module tripy)

U

unsqueeze() (in module tripy)

V

var() (in module tripy)

W

weight (tripy.BatchNorm attribute) (tripy.Conv attribute) (tripy.ConvTranspose attribute) (tripy.Embedding attribute) (tripy.GroupNorm attribute) (tripy.LayerNorm attribute) (tripy.Linear attribute)

weight_quant_dim (tripy.Linear attribute) weight_scale (tripy.Linear attribute) where() (in module tripy)

Z

zeros() (in module tripy)

zeros_like() (in module tripy)