Searched refs:sliceSizes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | optimize_mhlo.cc | 147 auto sliceSizes = gather.slice_sizes(); in matchAndRewrite() local
148 auto sliceSizesTy = sliceSizes.getType(); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorUtils.cpp | 105 SmallVector<int64_t, 4> sliceSizes(rank); in computeSliceSizes() local
107 sliceSizes[r] = std::min(sizes[r], shape[r] - elementOffsets[r]); in computeSliceSizes()
108 return sliceSizes; in computeSliceSizes()
|
| D | VectorOps.cpp | 961 auto sliceSizes = computeSliceSizes(shape, sizes, elementOffsets); in isValidExtractOrInsertSlicesType() local
964 VectorType::get(sliceSizes, vectorType.getElementType()); in isValidExtractOrInsertSlicesType()
1796 SmallVector<int64_t, 4> sliceSizes; in matchAndRewrite() local
1797 populateFromInt64AttrArray(extractStridedSliceOp.sizes(), sliceSizes); in matchAndRewrite()
1802 for (auto it : llvm::zip(maskDimSizes, sliceOffsets, sliceSizes)) { in matchAndRewrite()
|
| D | VectorTransforms.cpp | 214 auto sliceSizes = computeSliceSizes(shape, sizes, elementOffsets); in generateExtractSlicesOpResultType() local
216 vectorTypes[i] = VectorType::get(sliceSizes, vectorType.getElementType()); in generateExtractSlicesOpResultType()
1063 auto sliceSizes = computeSliceSizes(shape, sizes, elementOffsets); in matchAndRewrite() local
1066 loc, op.vector(), elementOffsets, sliceSizes, strides); in matchAndRewrite()
|