Lines Matching defs:resultShape

1070     ArrayRef<int64_t> resultShape = resultType.getShape();
1074     assert(resultShape.size() == currMixedSizes.size() &&
1076     for (auto it : llvm::zip(resultShape, currMixedSizes)) {
1332   SmallVector<int64_t> resultShape(indicesType.getShape().drop_back());
1333   resultShape.reserve(resultShape.size() + sourceType.getRank());
1337         resultShape.push_back(1);
1340     resultShape.push_back(sourceType.getDimSize(idx));
1342   return RankedTensorType::Builder(sourceType).setShape(resultShape);
3098                                         ArrayRef<int64_t> resultShape) {
3104   if (!resultShape.empty() && resultShape.size() != rank)
3111       inferredShape.push_back(resultShape.empty() ? ShapedType::kDynamic
3112                                                   : resultShape[i]);
3115       assert((resultShape.empty() || size == resultShape[i] ||
3116               resultShape[i] == ShapedType::kDynamic) &&
4216   SmallVector<int64_t> resultShape = llvm::to_vector(sourceShape);
4218     if (ShapedType::isDynamic(resultShape[tiledDim.value()]))
4221       resultShape[tiledDim.value()] = ShapedType::kDynamic;
4224     resultShape[tiledDim.value()] = divideCeilSigned(
4225         resultShape[tiledDim.value()], innerTileSizes[tiledDim.index()]);
4230     applyPermutationToVector(resultShape, outerDimsPerm);
4233   resultShape.append(innerTileSizes.begin(), innerTileSizes.end());
4234   return resultShape;
4280   SmallVector<int64_t> resultShape = getPackOpResultTypeShape(
4282   return RankedTensorType::get(resultShape, sourceType.getElementType());