FunctionsΒΆ
- all
- allclose
- any
- argmax
- argmin
- avgpool
- cast
- concatenate
- copy
- cos
- cumsum
- equal
- exp
- expand
- flatten
- flip
- gather
- gelu
- log
- masked_fill
- max
- maximum
- maxpool
- mean
- min
- minimum
- outer
- pad
- permute
- plugin
- prod
- relu
- repeat
- reshape
- resize
- rsqrt
- sigmoid
- silu
- sin
- softmax
- split
- sqrt
- squeeze
- stack
- sum
- tanh
- topk
- transpose
- unsqueeze
- var
- where