frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Mark aten.flip and aten.alias as core aten op (#93130)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93130
Approved by: https://github.com/qihqi, https://github.com/zhxchen17
This commit is contained in:
Sherlock Huang
2023-01-28 00:28:30 +00:00
committed by PyTorch MergeBot
parent 4d107e3426
commit b74a0fc486
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/native/native_functions.yaml
View File

@ -5682,6 +5682,7 @@
CPU, QuantizedCPU, CUDA, QuantizedCUDA: flip
MPS: flip_mps
autogen: flip.out
tags: core
- func: fliplr(Tensor self) -> Tensor
variants: function, method
@ -9676,6 +9677,7 @@
variants: method, function
dispatch:
CompositeExplicitAutograd: alias
tags: core
- func: _amp_foreach_non_finite_check_and_unscale_(Tensor(a!)[] self, Tensor(b!) found_inf, Tensor inv_scale) -> ()
variants: function