Lines Matching refs:shapesStorage
37 Operation &op, DenseMap<Value, ShapedTypeComponents> &shapesStorage) { in propagateShapesToTosaIf() argument
48 auto inferredTy = shapesStorage[op.getOperand(i)]; in propagateShapesToTosaIf()
75 Operation &op, DenseMap<Value, ShapedTypeComponents> &shapesStorage) { in propagateShapesToTosaWhile() argument
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()
267 for (auto it : shapesStorage) { in propagateShapesInRegion()