Lines Matching defs:srcShape

153   auto srcShape = cast<TensorType>(srcType).getShape();
156   if (srcShape.empty() || dstShape.empty())
159   if (ShapedType::isDynamicShape(srcShape) || ShapedType::isDynamicShape(dstShape)) {
162     for (auto i : llvm::seq<int64_t>(srcShape.size()))
169   while (currSrcDim < srcShape.size() && currDstDim < dstShape.size()) {
171     int64_t srcSize = srcShape[currSrcDim];
172     while (srcSize < dstSize && currSrcDim < srcShape.size()) {
175       srcSize *= srcShape[currSrcDim];
183         while (currSrcDim < srcShape.size() && srcShape[currSrcDim] == 1) {
195   assert(currSrcDim == srcShape.size() && currDstDim == dstShape.size());