mirror of https://github.com/llvm/torch-mlir
7750d2173a
Add argmax lowering from torch to linalg |
||
---|---|---|
.. | ||
argmax.py | ||
basic.py | ||
batchnorm.py | ||
conv.py | ||
elementwise.py | ||
main.py | ||
mlp.py | ||
quantized_models.py | ||
reduction.py | ||
vision_models.py | ||
xfail_sets.py |