torch-mlir/lib/Conversion/TorchToStablehlo
Zhekun Zhang 6a072d4f4a
[Stablehlo] AtenEmptyMemoryFormat remove device cpu check (#2288)
* remove cpu check

* update dtype

---------

Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-07-10 15:36:21 +08:00
..
Basic.cpp [Stablehlo] AtenEmptyMemoryFormat remove device cpu check (#2288) 2023-07-10 15:36:21 +08:00
CMakeLists.txt [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
GatherScatter.cpp [Stablehlo] Add `AtenIndexTensor` StableHlo support (#2107) 2023-05-24 11:13:57 -07:00
Linear.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +08:00
Pooling.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +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