bitorch_engine.layers.qconv.nbit.cutlass.layer

Classes

Q4Conv2dCutlass(*args, **kwargs)	A specialized 4-bit quantized convolutional layer using CUTLASS kernels.
Q4Conv2dCutlassForward(*args, **kwargs)	A custom forward function for 4-bit quantized convolution using CUTLASS kernels.