Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Tripy 0.0.3
Light Logo Dark Logo
Tripy 0.0.3

User Guides

  • An Introduction To Tripy
  • Quantization
  • Using the Compiler

API Reference

  • Compiling Code
    • compile
    • Executable
    • InputInfo
    • Stream
  • config
  • dtype
  • device
  • Exception
    • TripyException
  • logger
  • Module
    • Parameter
  • Operations
    • Functions
      • abs
      • all
      • allclose
      • any
      • argmax
      • argmin
      • avgpool
      • cast
      • concatenate
      • copy
      • cos
      • cumsum
      • exp
      • expand
      • flatten
      • flip
      • gather
      • gelu
      • log
      • masked_fill
      • max
      • maximum
      • maxpool
      • mean
      • minimum
      • outer
      • pad
      • permute
      • plugin
      • prod
      • relu
      • repeat
      • reshape
      • resize
      • rsqrt
      • sigmoid
      • silu
      • sin
      • softmax
      • split
      • sqrt
      • squeeze
      • stack
      • sum
      • tanh
      • transpose
      • unsqueeze
      • var
      • where
    • Initializers
      • arange
      • full
      • full_like
      • iota
      • iota_like
      • ones
      • ones_like
      • tril
      • triu
      • zeros
      • zeros_like
    • Modules
      • BatchNorm
      • Conv
      • ConvTranspose
      • Embedding
      • GroupNorm
      • LayerNorm
      • Linear
    • Quantization
      • dequantize
      • quantize
  • Tensor
    • DimensionSize
  • types

Developer Guides

  • Architecture
  • Debugging MLIR-TensorRT backend
  • Design Decisions
  • Adding New Operators
  • Using Python APIs of MLIR Dialects
Back to top
View this page

InitializersΒΆ

  • arange
  • full
  • full_like
  • iota
  • iota_like
  • ones
  • ones_like
  • tril
  • triu
  • zeros
  • zeros_like
Next
arange
Previous
where
Copyright © 2024, NVIDIA
Made with Sphinx and @pradyunsg's Furo