We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9adad9b commit 80bd093Copy full SHA for 80bd093
include/torch-mlir/Conversion/TorchOnnxToTorch/Patterns.h
@@ -95,6 +95,16 @@ struct OpBinder {
95
return success();
96
}
97
98
+ ParseResult tensorResultTypeAtIndex(Torch::ValueTensorType &typeIdx, int64_t idx) {
99
+ if (idx >= op->getNumResults())
100
+ return failure();
101
+ auto t = toValidTensorType(op->getResult(idx).getType());
102
+ if (!t)
103
104
+ typeIdx = t;
105
+ return success();
106
+ }
107
+
108
// Attribute accessors.
109
ParseResult s64BoolAttr(bool &value, StringRef nameSuffix,
110
bool defaultValue = false) {
0 commit comments