llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added tensorResultTypeAtIndex to Patterns.h 

Need this for LayerNorm
pull/2722/head
Andreas Falkenberg 2023-12-29 11:21:55 -08:00 committed by Vivek Khandelwal
parent 9adad9bc40
commit 80bd093d56
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
include/torch-mlir/Conversion/TorchOnnxToTorch/Patterns.h
View File

@ -95,6 +95,16 @@ struct OpBinder {
return success(); return success();
} }
ParseResult tensorResultTypeAtIndex(Torch::ValueTensorType &typeIdx, int64_t idx) {
if (idx >= op->getNumResults())
return failure();
auto t = toValidTensorType(op->getResult(idx).getType());
if (!t)
return failure();
typeIdx = t;
return success();
}
// Attribute accessors. // Attribute accessors.
ParseResult s64BoolAttr(bool &value, StringRef nameSuffix, ParseResult s64BoolAttr(bool &value, StringRef nameSuffix,
bool defaultValue = false) { bool defaultValue = false) {