Wu Yuan cc28d566ff [Stablehlo] Support AtenTrilOp (#3359) ... 1. lower aten.tril to stablehlo composed by iota, select and so forth 2. add related e2e test cases		2024-05-20 15:49:24 +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	[Stablehlo]index type use i64 (#3354)	2024-05-16 15:33:23 +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 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