Home
last modified time | relevance | path

Searched refs:slicedDimensions (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/include/mlir/Dialect/Tensor/Transforms/
H A DTransformUtils.h172 const llvm::SmallBitVector &slicedDimensions, ArrayRef<Value> tiledSizes) in ExtractSliceFromCollapseHelper() argument
178 slicedDimensions(slicedDimensions), tiledSizes(tiledSizes) {} in ExtractSliceFromCollapseHelper()
203 llvm::SmallBitVector slicedDimensions; variable
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DExtractSliceFromReshapeUtils.cpp102 llvm::SmallBitVector slicedDimensions = in create() local
110 if (slicedDimensions[i] && linearizedDimensions[i]) in create()
117 linearizedDimensions, slicedDimensions, tileSizes); in create()
135 if (linearizedDimensions[i] && slicedDimensions[i]) { in emitLoopNestBody()
/llvm-project/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h473 slicedDimensions(getSlicedDimensions(collapseShapeOutputShape,
503 llvm::SmallBitVector slicedDimensions;
501 llvm::SmallBitVector slicedDimensions; global() variable
/llvm-project/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp305 if (slicedDimensions[it.index()] && linearizedDimensions[it.index()]) { in getExtractSliceParams()
341 if (linearizedDimensions[i] && slicedDimensions[i]) { in getInsertSliceParams()