Jiawei Wu edc87fc577 [stablehlo] support dynamic-shaped index in stablehlo conversion for aten.index-like ops (#3322) ... For now, at most one dynamic dim of index tensors in aten.index/aten.index_put-like op is supported.		2024-08-01 10:41:09 +08:00
..
Basic.cpp	[stablehlo] support dynamic-shaped index in stablehlo conversion for aten.index-like ops (#3322)	2024-08-01 10:41:09 +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] support dynamic-shaped index in stablehlo conversion for aten.index-like ops (#3322)	2024-08-01 10:41:09 +08:00
Linear.cpp	[Stablehlo] simplify promoteType (#3525)	2024-07-10 10:52:19 +08:00
Pooling.cpp	[Stablehlo] simplify promoteType (#3525)	2024-07-10 10:52:19 +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] use index type as dim size, avoid to generate index_cast (#3526)	2024-07-07 18:03:03 +08:00
Rng.cpp	[NFC] Remove unused header files (#3386)	2024-05-30 14:30:36 +08:00
StablehloLegalizeUtils.cpp	[stablehlo] support dynamic-shaped index in stablehlo conversion for aten.index-like ops (#3322)	2024-08-01 10:41:09 +08:00
TorchToStablehlo.cpp	[Stablehlo] support uint8 (#3367)	2024-06-04 09:04:59 +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] use index type as dim size, avoid to generate index_cast (#3526)	2024-07-07 18:03:03 +08:00