Lines Matching refs:getShape

129   auto shape = getShape();  in getNumElements()
136 int64_t ShapedType::getRank() const { return getShape().size(); } in getRank()
142 return getShape()[i]; in getDimSize()
148 return llvm::count_if(getShape().take_front(index), ShapedType::isDynamic); in getDynamicDimIndex()
170 ArrayRef<int64_t> ShapedType::getShape() const { in getShape() function in ShapedType
173 return cast<VectorType>().getShape(); in getShape()
175 return cast<RankedTensorType>().getShape(); in getShape()
177 return cast<MemRefType>().getShape(); in getShape()
184 return llvm::count_if(getShape(), isDynamic); in getNumDynamicDims()
188 return hasRank() && llvm::none_of(getShape(), isDynamic); in hasStaticShape()
192 return hasStaticShape() && getShape() == shape; in hasStaticShape()
228 ArrayRef<int64_t> VectorType::getShape() const { return getImpl()->getShape(); } in getShape() function in VectorType
271 ArrayRef<int64_t> RankedTensorType::getShape() const { in getShape() function in RankedTensorType
272 return getImpl()->getShape(); in getShape()
380 ArrayRef<int64_t> MemRefType::getShape() const { return getImpl()->getShape(); } in getShape() function in MemRefType
553 makeCanonicalStridedLayoutExpr(t.getShape(), t.getContext()); in getStridesAndOffset()
721 makeCanonicalStridedLayoutExpr(t.getShape(), t.getContext()); in canonicalizeStridedLayout()
726 return MemRefType::get(t.getShape(), t.getElementType(), in canonicalizeStridedLayout()
730 return MemRefType::get(t.getShape(), t.getElementType(), {}); in canonicalizeStridedLayout()