pytorch/torch/testing
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
..
_internal prod ref 2022-05-31 14:18:49 +00:00
__init__.py cleanup torch.testing namespace (#72708) 2022-02-25 06:30:31 +00:00
_comparison.py Fix a few issues on assert/double error/legacy constructor (#77966) 2022-05-20 20:25:12 +00:00
_creation.py [chalf] div, eq, masked_fill, index_put (#77479) 2022-05-18 17:01:08 +00:00
_deprecated.py promote torch.testing to stable (#73348) 2022-02-25 06:30:31 +00:00
_legacy.py [complex32] real and imag (also remove unused real and imag kernels) 2022-05-05 04:36:58 +00:00