026e8db2e4
[Stablehlo] add converter for aten.scatter.src op ( #2295 )
2023-07-24 10:14:45 +08:00
64d7626a52
Fixes for split tensor and slice ( #2314 )
...
* RecomposeComplexOps: Remove dead slice op
* lib/Dialect/Torch/IR/TorchOps.cpp: Fold slice ops even when they are on non-value tensors
* lib/Conversion/TorchToTosa/TorchToTosa.cpp: Fix slice start/end out of range/none
* lib/Dialect/Torch/IR/TorchOps.cpp: AtenSliceTensorOp::fold: Fold slices that go from 0:int_max
* More tests for aten.split.Tensor
2023-07-20 09:53:54 +02:00
f0b7b63be0
[Stablehlo] Add aten.uniform lowering ( #2101 )
...
* add uniform stablehlo lowering
* add unit test
* new line
* rm redundant file
* Empty commit, trigger test
* fix include
* address comments
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-25 10:32:55 +08:00
51e0a2c933
[Stablehlo] Add stablehlo support for aten.abs ( #2068 )
...
Co-authored-by: AmosLewis <Amos_Lewsi@foxmail.com>
2023-05-08 22:13:00 -07:00
0096ceae2f
[Stablehlo] fix reduce max init_value with -inf ( #2064 )
...
* [Stablehlo] fix reduce max init_value with -inf
* update
2023-05-06 12:05:51 -07:00
1bd5747ca3
[StableHlo] Fix transposed convolution conversion ( #2026 )
...
* fix conv bwd
* fix
* fix group case
* clean up
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-04-13 11:24:39 -07:00
c86f46bd70
[test] rename TorchToMhlo to TorchToStablehlo ( #1995 )
2023-04-03 18:41:25 -07:00
Jiawei Wu
Matthias Gehre
Zhekun Zhang
Chi_Liu
Yuanqiang Liu