Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp210 SmallVector<ShapedTypeComponents> returnedShapes; in propagateShapesInRegion() local
216 op.getRegions(), returnedShapes) in propagateShapesInRegion()
218 for (auto it : llvm::zip(op.getResults(), returnedShapes)) { in propagateShapesInRegion()
H A DTosaDecomposeTransposeConv.cpp46 SmallVector<ShapedTypeComponents> returnedShapes; in createOpAndInfer() local
50 op->getRegions(), returnedShapes) in createOpAndInfer()
59 auto predictedShape = returnedShapes[0]; in createOpAndInfer()