Yuanqiang Liu 5928f68e60 [Stablehlo] refactor amax, max, max.dim's lowering to stablehlo (#3348) ... * not to decompose `aten.amax` on `stablehlo` backend. Because it could be lowering to `stablehlo.reduce` directly. * lowering `aten.max.dim` to `stablehlo.reduce apply max` when `AtenMaxDimOp.getIndices()` doesn't have users. It's more simple.		2024-05-16 00:05:19 +08:00
..
Basic.cpp	[Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328)	2024-05-11 15:33:37 +08:00
CMakeLists.txt	[Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328)	2024-05-11 15:33:37 +08:00
GatherScatter.cpp	Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243)	2024-04-27 14:00:56 -07:00
Linear.cpp	Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243)	2024-04-27 14:00:56 -07:00
Pooling.cpp	[Stablehlo] Improve the lowering of pool op in stablehlo (#3259)	2024-05-01 00:06:13 +08:00
PopulatePatterns.h	[Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328)	2024-05-11 15:33:37 +08:00
Reduction.cpp	[Stablehlo] refactor amax, max, max.dim's lowering to stablehlo (#3348)	2024-05-16 00:05:19 +08:00
Rng.cpp	[Stablehlo] fix aten.randn's lowering with f32 element type (#3329)	2024-05-11 17:40:04 +08:00
StablehloLegalizeUtils.cpp	[stablehlo] Support PrimsCollapseOp and PrimsSplitDimOp in stablehlo (#3230)	2024-04-29 17:40:30 +08:00
TorchToStablehlo.cpp	[Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328)	2024-05-11 15:33:37 +08:00
Utils.cpp	[Stablehlo] add torch_to_stablehlo::getBackendTypeForScalarType (#2975)	2024-03-04 23:31:54 +08:00
Utils.h	[Stablehlo] add torch_to_stablehlo::getBackendTypeForScalarType (#2975)	2024-03-04 23:31:54 +08:00
ViewLike.cpp	[stablehlo] Support PrimsCollapseOp and PrimsSplitDimOp in stablehlo (#3230)	2024-04-29 17:40:30 +08:00