Bhargav Yagnik a098bc92e0 Fix: Preserve input dtype in Dropout layer output (#1323) ... * Fix: Preserve input dtype in Dropout layer output - Modified Dropout implementation to ensure that the output dtype matches the input dtype. - This resolves the issue #1321 * Update test cases in test_nn.py - Revised test cases to align with updated dropout code - Fixed assertion method: replaced self.assertTrue with self.assertEqual for accurate comparisons in test_nn.py -> test_rope, test_alibi and test_dropout, * updated dropout.py		2024-08-13 11:54:21 -07:00
..
mlx	Fix: Preserve input dtype in Dropout layer output (#1323)	2024-08-13 11:54:21 -07:00
src	Add memory_efficient_threshold kwarg to sdpa kernel (#1319)	2024-08-12 12:57:09 -07:00
tests	Fix: Preserve input dtype in Dropout layer output (#1323)	2024-08-13 11:54:21 -07:00