Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp37 Operation &op, DenseMap<Value, ShapedTypeComponents> &shapesStorage) { in propagateShapesToTosaIf() argument
48 auto inferredTy = shapesStorage[op.getOperand(i)]; in propagateShapesToTosaIf()
86 auto shapedTypeComponent = shapesStorage[operand]; in propagateShapesToTosaWhile()
182 DenseMap<Value, ShapedTypeComponents> shapesStorage; in propagateShapesInRegion() local
185 shapesStorage[val] = st; in propagateShapesInRegion()
187 shapesStorage[val] = t; in propagateShapesInRegion()
191 auto it = shapesStorage.find(val); in propagateShapesInRegion()
192 if (it == shapesStorage.end()) in propagateShapesInRegion()
202 propagateShapesToTosaIf(op, shapesStorage); in propagateShapesInRegion()
203 propagateShapesToTosaWhile(op, shapesStorage); in propagateShapesInRegion()
[all …]