madt2709 4bb98d8882 Fix track_running_stats in batchnorm (#6200) ... * Fix track_running_stats in batchnorm * Fix linter * Update test_fold_conv_batchnorm_notrain to keep allowed at 1 * Add test_fold_conv_batchnorm_notrain_no_running_stats * Save 1 line		2024-08-20 14:01:22 -07:00
..
codegen	fix indexing out of bounds (#6208)	2024-08-20 11:34:56 -07:00
engine	hotfix: more lazyop rename to uop [run_process_replay] (#6157)	2024-08-18 17:28:44 +03:00
nn	Fix track_running_stats in batchnorm (#6200)	2024-08-20 14:01:22 -07:00
renderer	Refactor amd kernel prefix (#6205)	2024-08-20 10:37:36 -07:00
runtime	_gpu_alloc -> allocator.alloc (#6189)	2024-08-19 23:34:22 +03:00
shape	re:6125 switch real_size to use uops [run_process_replay] (#6138)	2024-08-19 13:20:24 -04:00
__init__.py	add mnist tutorial	2024-04-25 16:08:32 +08:00
device.py	RUF018 assignment-in-assert [run_process_replay] (#6172)	2024-08-19 00:34:52 -04:00
dtype.py	threefry touchup [run_process_replay] (#6169)	2024-08-18 23:01:24 -04:00
function.py	remove UnaryOps.NEG from function.py (#6187)	2024-08-19 17:39:15 -04:00
helpers.py	Support gunzip in fetch (#6176)	2024-08-19 12:04:40 -07:00
lazy.py	remove UnaryOps.NEG from lazy.py (#6193)	2024-08-19 18:41:28 -04:00
multi.py	MLB support reshape for uneven shards (#5804)	2024-08-01 02:36:03 -07:00
ops.py	type verify intermediate UOps [run_process_replay] (#6140)	2024-08-19 20:59:01 +03:00
py.typed	add a single py.typed (#6083)	2024-08-14 17:31:46 -07:00
tensor.py	remove UnaryOps.NEG from function.py (#6187)	2024-08-19 17:39:15 -04:00