Lines Matching defs:castOp
55 auto castOp = cast<tensor::CastOp>(op);
57 castOp.getSource(), options, invocationStack);
66 if (isa<UnrankedTensorType>(castOp.getSource().getType())) {
69 return getMemRefTypeWithFullyDynamicLayout(castOp.getType(), memorySpace);
73 if (isa<UnrankedTensorType>(castOp.getType())) {
74 return getMemRefTypeWithFullyDynamicLayout(castOp.getType(), memorySpace);
79 auto rankedResultType = cast<RankedTensorType>(castOp.getType());
87 auto castOp = cast<tensor::CastOp>(op);
91 getBuffer(rewriter, castOp.getSource(), options);
97 bufferization::getBufferType(castOp.getResult(), options);