Home
last modified time | relevance | path

Searched refs:permsShape (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp648 ShapeAdaptor permsShape = operands.getShape(1); in inferReturnTypeComponents() local
652 if (!inputShape.hasRank() || !permsShape.hasRank() || in inferReturnTypeComponents()
653 permsShape.isDynamicDim(0)) { in inferReturnTypeComponents()
660 if (permsShape.getDimSize(0) != inputShape.getRank()) { in inferReturnTypeComponents()
668 outputShape.resize(permsShape.getDimSize(0), ShapedType::kDynamicSize); in inferReturnTypeComponents()