Lines Matching refs:staticSizes
1193 SmallVector<int64_t, 4> staticSizes; in build() local
1194 dispatchIndexOpFoldResults(sizes, dynamicSizes, staticSizes, in build()
1196 auto resultType = RankedTensorType ::get(staticSizes, elementType); in build()
1197 build(b, result, resultType, dynamicSizes, b.getI64ArrayAttr(staticSizes)); in build()
1203 SmallVector<int64_t, 4> staticSizes = llvm::to_vector<4>(llvm::map_range( in verify() local
1216 staticSizes, resultType.getElementType(), resultType.getEncoding()); in verify()
1225 Type InitTensorOp::inferResultType(ArrayRef<int64_t> staticSizes, in inferResultType() argument
1227 return RankedTensorType::get(staticSizes, elementType, encoding); in inferResultType()
1263 SmallVector<int64_t, 4> staticSizes; in matchAndRewrite() local
1267 staticSizes.push_back(op.getStaticSize(i)); in matchAndRewrite()
1277 staticSizes.push_back(constantIndexOp.value()); in matchAndRewrite()
1283 staticSizes.push_back(ShapedType::kDynamicSize); in matchAndRewrite()
1286 RankedTensorType::get(staticSizes, op.getType().getElementType()); in matchAndRewrite()
1291 rewriter.getI64ArrayAttr(staticSizes)); in matchAndRewrite()