pytorch/torch/_decomp
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
..
__init__.py [core IR] Add a core decomposition for aten.all (#110093) 2023-09-27 01:31:41 +00:00
decompositions_for_jvp.py Fix registering jit decompositions for jvp for out wrapped decomps (#109367) 2023-09-21 16:36:51 +00:00
decompositions_for_rng.py Implement decomposition for aten.rrelu_with_noise (#106812) 2023-08-11 19:18:29 +00:00
decompositions.py [core IR] Add a core decomposition for floor_divide (#110046) 2023-09-26 08:39:21 +00:00