Bitorch Engine
Contents:
Installation
Build options
Full Documentation
bitorch_engine
bitorch_engine.functions
bitorch_engine.layers
bitorch_engine.layers.qconv
bitorch_engine.layers.qembedding
bitorch_engine.layers.qlinear
bitorch_engine.layers.qmha
bitorch_engine.optim
bitorch_engine.utils
Bitorch Engine
Full Documentation
bitorch_engine
bitorch_engine.layers
bitorch_engine.layers.qconv
bitorch_engine.layers.qconv.binary
bitorch_engine.layers.qconv.binary.cutlass
bitorch_engine.layers.qconv.binary.cutlass.extension
View page source
bitorch_engine.layers.qconv.binary.cutlass.extension
Functions
get_ext
Obtains the CUDA extension for Cutlass-based binary convolution.