colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix operand type access.

This commit is contained in:
Doru Bercea 2020-01-20 15:48:16 -05:00
parent bd44d8402e
commit 6b55bb43c7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/dialect/onnx/onnx_ops.cpp
View File

@ -408,7 +408,7 @@ void ONNXTransposeOp::inferShapes() {
// Naive transposition which handles the default case of // Naive transposition which handles the default case of
// reversing the shape of the tensor (similar to numpy.transpose). // reversing the shape of the tensor (similar to numpy.transpose).
auto arrayTy = getOperand()->getType().cast<RankedTensorType>(); auto arrayTy = getOperand().getType().cast<RankedTensorType>();
SmallVector<int64_t, 2> dims; SmallVector<int64_t, 2> dims;
if (auto permutation = getAttrOfType<ArrayAttr>( if (auto permutation = getAttrOfType<ArrayAttr>(
@ -422,7 +422,7 @@ void ONNXTransposeOp::inferShapes() {
dims.emplace_back(dim); dims.emplace_back(dim);
} }
getResult()->setType(RankedTensorType::get(dims, arrayTy.getElementType())); getResult().setType(RankedTensorType::get(dims, arrayTy.getElementType()));
} }
LogicalResult verify(ONNXTransposeOp op) { LogicalResult verify(ONNXTransposeOp op) {