Cheng 5be5daa6ef Use compiled function in Sigmoid module (#1116)		2024-05-14 06:25:57 -07:00
..
__init__.py	Conv3d (#993)	2024-05-11 06:15:02 -07:00
activations.py	Use compiled function in Sigmoid module (#1116)	2024-05-14 06:25:57 -07:00
base.py	Support Chaining for some of functionalities of nn.Module (#885) (#897)	2024-03-27 19:58:29 -07:00
containers.py	copyright + ack	2023-11-30 11:12:53 -08:00
convolution.py	Conv3d (#993)	2024-05-11 06:15:02 -07:00
dropout.py	Make shape a tuple (#591)	2024-01-30 13:11:01 -08:00
embedding.py	Quantize embedding (#994)	2024-04-15 16:42:10 -07:00
linear.py	Reduce a little overhead (#871)	2024-03-22 17:29:36 -07:00
normalization.py	Reduce a little overhead (#871)	2024-03-22 17:29:36 -07:00
pooling.py	avgpool, not maxpool (#1002)	2024-04-17 08:26:22 -07:00
positional_encoding.py	No reshape rope (#838)	2024-03-18 17:03:07 -07:00
quantized.py	Use correct module type in quantized.py (#1115)	2024-05-14 06:25:42 -07:00
recurrent.py	fix a couple bugs (#952)	2024-04-02 12:07:41 -07:00
transformer.py	Change the transformer to norm_first by default (#599)	2024-01-31 12:55:30 -08:00
upsample.py	Upsample with bicubic interpolation (#967)	2024-04-10 15:47:22 -07:00