Yuanqiang Liu 5bb1a65ec9 [Stablehlo] refactor reduction lowering and support aten.amin (#3383) ... * implement detailed lowering template pattern `ConvertAtenReduceAllDimsOp` and `ConvertAtenReduceKeepDimOp` * support `aten.amin`'s lowering.		2024-05-23 20:40:20 +08:00
..
Basic.cpp	[Stablehlo] Support AtenTrilOp (#3359)	2024-05-20 15:49:24 +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	Bump llvm and stablehlo (#3377)	2024-05-22 23:28:45 +08: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 reduction lowering and support aten.amin (#3383)	2024-05-23 20:40:20 +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	Bump llvm and stablehlo (#3377)	2024-05-22 23:28:45 +08:00