pytorch/torch/_refs
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
..
nn Add Elementwise unary ops 4 references (#78216) 2022-05-27 21:55:34 +00:00
special Add i0, i1, zeta refs (#78111) 2022-05-23 21:33:56 +00:00
__init__.py prod ref 2022-05-31 14:18:49 +00:00