llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion/TorchToStablehlo
History
JianzheXiao 31ef08b63d
[Stablehlo]Add support for AvgPool1dOp (#2268) 
* Add support for AvgPool1d

* Update AbstractInterpLibrary

* support avgpool1d in linalg

* refactored code

* fix nit problem
 		2023-07-25 14:09:53 +08:00
..
Basic.cpp [Torch-Dialect] emit aten.narrow.Tensor op and decompose it to aten.narrow op (#2297) 2023-07-20 16:46:44 +08:00
CMakeLists.txt [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
GatherScatter.cpp [Stablehlo] add converter for aten.scatter.src op (#2295) 2023-07-24 10:14:45 +08:00
Linear.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +08:00
Pooling.cpp [Stablehlo]Add support for AvgPool1dOp (#2268) 2023-07-25 14:09:53 +08:00
PopulatePatterns.h [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
Reduction.cpp Bump LLVM (#2176) 2023-06-13 16:17:23 +02:00
StablehloLegalizeUtils.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +08:00
TorchToStablehlo.cpp [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
ViewLike.cpp [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30