Searched refs:getIndexTensorAttr (Results 1 – 4 of 4) sorted by relevance
667 return builder.getIndexTensorAttr(resultShape); in fold()766 rewriter.getIndexTensorAttr(foldedConstantShape))); in matchAndRewrite()862 return builder.getIndexTensorAttr(resultShape); in fold()899 result.addAttribute("shape", builder.getIndexTensorAttr(ints)); in parse()1155 return builder.getIndexTensorAttr(extents); in fold()1562 return builder.getIndexTensorAttr(type.getShape()); in fold()1725 results.push_back(builder.getIndexTensorAttr(shape.take_front(splitPoint))); in fold()1726 results.push_back(builder.getIndexTensorAttr(shape.drop_front(splitPoint))); in fold()
138 DenseIntElementsAttr getIndexTensorAttr(ArrayRef<int64_t> values);
141 DenseIntElementsAttr Builder::getIndexTensorAttr(ArrayRef<int64_t> values) { in getIndexTensorAttr() function in Builder
1519 coord = mlir::Builder(parser.getContext()).getIndexTensorAttr(zippedBounds); in parseCustomRangeSubscript()