Searched refs:unrolledShape (Results 1 – 1 of 1) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 224 SmallVector<int64_t, 4> unrolledShape; member
239 state.unrolledShape.resize(vectorType.getRank()); in initUnrolledVectorState()
240 getMappedElements(indexMap, targetShape, state.unrolledShape); in initUnrolledVectorState()
243 shapeRatio(vectorType.getShape(), state.unrolledShape); in initUnrolledVectorState()
252 SmallVector<int64_t, 4> sizes(state.unrolledShape); in initUnrolledVectorState()
268 SmallVector<int64_t, 4> sliceOffsets(state.unrolledShape.size()); in getUnrolledVectorLinearIndex()
282 SmallVector<int64_t, 4> sliceOffsets(state.unrolledShape.size()); in getOrCreateUnrolledVectorSlice()
285 SmallVector<int64_t, 4> sliceStrides(state.unrolledShape.size(), 1); in getOrCreateUnrolledVectorSlice()
296 VectorType::get(state.unrolledShape, initValueType.getElementType()); in getOrCreateUnrolledVectorSlice()
397 auto unrolledResultType = VectorType::get(resultValueState.unrolledShape, in unrollSingleResultStructuredOp()
[all …]
|