llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion/TorchToMhlo
History
武家伟 636f5acb10
[MHLO] Init MHLO reduce-like op conversion (#1133) 
* [MHLO] init reduce-like op conversion from Torch to MHLO
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
 		2022-08-03 10:47:52 +08:00
..
BasicOp.cpp [MHLO] support non-constant torch scalar in BasicOps (#1134) 2022-08-03 08:16:31 +08:00
CMakeLists.txt [MHLO] Init MHLO reduce-like op conversion (#1133) 2022-08-03 10:47:52 +08:00
GatherOp.cpp [MHLO] Init MHLO gather op patterns (#1104) 2022-07-25 23:47:46 +08:00
MhloLegalizeUtils.cpp [MHLO] support non-constant torch scalar in BasicOps (#1134) 2022-08-03 08:16:31 +08:00
MhloLegalizeUtils.h [MHLO] support non-constant torch scalar in BasicOps (#1134) 2022-08-03 08:16:31 +08:00
PopulatePatterns.h [MHLO] Init MHLO reduce-like op conversion (#1133) 2022-08-03 10:47:52 +08:00
ReductionOp.cpp [MHLO] Init MHLO reduce-like op conversion (#1133) 2022-08-03 10:47:52 +08:00
TorchToMhlo.cpp [MHLO] Init MHLO reduce-like op conversion (#1133) 2022-08-03 10:47:52 +08:00
ViewLikeOps.cpp [NFC][MHLO] move util funcs to MhloLegalizeUtils.h/cpp (#1128) 2022-08-02 09:21:37 +08:00