mirror of https://github.com/llvm/torch-mlir
f97cd4893f
Shapes can be processed as tensors to represent the set of dimensions. As reshapes take a list of scalars this can result in a single dynamic dimension blocking the adjacent static dimensions. This pass attempts to de-couple tensor computations related to shapes and propagate values to better support lowering scalar tensor computations. |
||
|---|---|---|
| .. | ||
| _torch_mlir_custom_op_example | ||
| csrc | ||
| jit_ir_importer | ||
| _dynamo_fx_importer.py | ||
| _version.py | ||
| compiler_utils.py | ||
| dynamo.py | ||
| torchscript.py | ||