llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test/Dialect/Torch
History
Prashant Kumar 68acc8696e Modify softmax decomposition to be more numerically stable. 
The softmax decomposition is modified according to https://github.com/pytorch/functorch/blob/main/functorch/_src/decompositions.pytorch
to account for numerical stability. Also, modified aten.argmax lowering
to handle negative dimension.
 		2022-02-03 21:20:36 +05:30
..
GlobalizeObjectGraph Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
adjust-calling-conventions.mlir Add support for multiple return values 2021-11-16 21:07:45 -05:00
canonicalize.mlir Fold __is__ and unchecked_cast of derefine 2022-01-28 17:54:40 -05:00
decompose-complex-ops.mlir Modify softmax decomposition to be more numerically stable. 2022-02-03 21:20:36 +05:30
inline-global-slots.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
invalid.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
maximize-value-semantics.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
ops.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
prepare-for-globalize-object-graph.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
promote-types.mlir Make broadcasting result shape more static 2022-01-06 18:39:27 -05:00
reduce-op-variants-error.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
reduce-op-variants.mlir [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
refine-public-return.mlir Add support for multiple return values 2021-11-16 21:07:45 -05:00
refine-types-branch.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
refine-types.mlir RefineTypes fixes for TOSA backend (#557) 2022-02-01 14:08:54 -08:00