llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/e2e_testing/torchscript
History
Prashant Kumar ab81f871e4 Add aten.tensor.int and aten.tensor.float op lowerings. 
Add the required lowerings and correct test cases.
These op produce zero-d tensors and it was incorrectly mentioned in
refine types to produce 1d tensor of size 1.
 		2021-12-15 17:21:34 +05:30
..
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
backprop.py Add e2e test for aten.log_softmax_back_data op 2021-11-19 00:08:28 +05:30
basic.py Add aten.tensor.int and aten.tensor.float op lowerings. 2021-12-15 17:21:34 +05:30
batchnorm.py [MLIR][TORCH] Add E2E support for `aten.native_layer_norm`. (#470) 2021-12-10 19:06:19 +05:30
conv.py Linalg lowering for aten.conv2d(bias=True) 2021-12-08 14:44:36 -08:00
elementwise.py Add support for int types in gtScalar op. 2021-12-14 01:29:52 +05:30
index_select.py [MLIR][TORCH] Add E2E support for aten.index_select op 2021-12-09 23:13:36 +05:30
main.py [MLIR][TORCH] Add E2E support for aten.index_select op 2021-12-09 23:13:36 +05:30
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Support aten::linear with rank 3 inputs 2021-11-18 22:15:04 +05:30
nll_loss.py Add aten::nll_loss_forward op lowering. 2021-12-07 17:11:08 +05:30
quantized_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reduction.py [MLIR][TORCH] Add E2E support for aten.mean and aten.numel op. 2021-12-02 11:51:13 +05:30
scalar.py add aten.add.int lowering in TorchToStd 2021-11-29 13:22:50 -05:00
slice_like.py Add lowering for slice and selectInt (#398) 2021-12-02 22:09:21 -06:00
squeeze.py [TORCH][MLIR] Add E2E support for `aten.squeeze.dim` op 2021-12-10 17:01:20 +05:30
type_conversion.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
type_promotion.py Add scalar type promotion for mul and div (#454) 2021-12-03 13:51:25 -06:00
view.py [MLIR][TORCH] Add E2E support for `torch.aten.view` 2021-10-29 22:33:10 +05:30
vision_models.py Add min/max/clamp support. 2021-10-27 13:29:21 -07:00
xfail_sets.py * [tosa] Support for Rsqrt legalization (#480) 2021-12-14 10:03:58 -08:00