Edward Z. Yang b7215de32f prod ref ... It turns out the prim is implemented incorrectly as torch.prod does not accept a dim list, so I added a little stub for this. Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/78461 Approved by: https://github.com/ngimel		2022-05-31 14:18:49 +00:00
..
__init__.py	prod ref	2022-05-31 14:18:49 +00:00
context.py	Add logit ref; allow non-refs to be called in refs.	2022-05-21 02:35:14 +00:00
executor.py	Apply black lint CI to PrimTorch	2022-05-11 18:23:12 +00:00
utils.py	addr ref	2022-05-25 01:40:11 +00:00
wrappers.py	[primTorch] Adds random operations (#78026)	2022-05-23 01:56:28 +00:00