Angelos Katharopoulos 4847199ec6 Add the quantization type option to quantizable layers		2024-12-16 22:11:23 -08:00
..
__init__.py	Adds 3D pooling (#1526)	2024-11-19 16:45:24 -08:00
activations.py	feat: Added "tanh" option to GELU approximation (#1268)	2024-07-28 09:07:56 +02:00
base.py	fix module attribute set, reset, set (#1403)	2024-09-11 16:30:42 -07:00
containers.py	copyright + ack	2023-11-30 11:12:53 -08:00
convolution_transpose.py	Transposed Convolution (#1245)	2024-09-06 19:52:38 -07:00
convolution.py	add groups in conv2d (#1569)	2024-11-07 13:57:53 -08:00
dropout.py	Typing the dropout. (#1479)	2024-10-15 06:45:46 -07:00
embedding.py	Add the quantization type option to quantizable layers	2024-12-16 22:11:23 -08:00
linear.py	Add the quantization type option to quantizable layers	2024-12-16 22:11:23 -08:00
normalization.py	faster group norm (#1304)	2024-08-01 12:49:23 -07:00
pooling.py	Doc fix (#1615)	2024-11-22 11:12:25 -08:00
positional_encoding.py	Doc error for default for scale in SinusoidalPositionalEncoding (#1174)	2024-06-02 13:42:45 -07:00
quantized.py	Add the quantization type option to quantizable layers	2024-12-16 22:11:23 -08:00
recurrent.py	Faster RNN layers (#1419)	2024-09-17 06:04:19 -07:00
transformer.py	fix transformer decoder post norm LN (#1637)	2024-12-02 07:02:17 -08:00
upsample.py	improvements to scatter / gather (#1541)	2024-10-30 19:30:54 -07:00