pytorch/torch/_refs
SS-JIA dec140f1ea [core IR] Add a core decomposition for aten.all (#110093)
## Context

Change the ref implementation of `aten.all` to only use other `torch` operators such that we can use it for the core ATen decomposition table. This will replace the decomposition for `aten.all` that was used specifically by Inductor.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110093
Approved by: https://github.com/manuelcandales, https://github.com/peterbell10, https://github.com/lezcano
2023-09-27 01:31:41 +00:00
..
linalg Add reference for linalg.vecdot (#108188) 2023-08-31 15:30:23 +00:00
nn Decomp for aten.dropout (#106274) 2023-08-23 21:12:37 +00:00
special Replace _prims_common.check with torch._check* (#103240) 2023-06-21 00:46:17 +00:00
__init__.py [core IR] Add a core decomposition for aten.all (#110093) 2023-09-27 01:31:41 +00:00
_conversions.py Replace _prims_common.check with torch._check* (#103240) 2023-06-21 00:46:17 +00:00
fft.py decomposition spectral ops fixes (#108360) 2023-09-09 04:48:09 +00:00