Searched refs:operandShape (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | NvGpuSupport.cpp | 44 std::array<int64_t, 2> getTileShape(ArrayRef<int64_t> operandShape, in getTileShape() argument
47 return {operandShape[0] / kNumRowsPerTile, in getTileShape()
48 (operandShape[1] * elementType.getIntOrFloatBitWidth()) / in getTileShape()
156 ArrayRef<int64_t> operandShape, in getRegisterIndexToTileOffsetMap() argument
163 getTileShape(operandShape, elementType, lineSize); in getRegisterIndexToTileOffsetMap()
176 ArrayRef<int64_t> operandShape = fragmentType.vectorType.getShape(); in getLaneIdAndValueIdToOperandCoord() local
193 lineSize, elementType, operandShape, in getLaneIdAndValueIdToOperandCoord()
245 ArrayRef<int64_t> operandShape = params.fragmentType.getShape(); in getLaneIdToLdMatrixMatrixCoord() local
254 AffineExpr row = d0 % (operandShape[0]); in getLaneIdToLdMatrixMatrixCoord()
255 AffineExpr col = d0.floorDiv(operandShape[0]) * (kElementsPer128b); in getLaneIdToLdMatrixMatrixCoord()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | InferTypeOpInterface.h | 168 ValueShapeRange(ValueRange values, ValueShapeMapFn operandShape = nullptr,
170 : RangeBaseT(values), operandShape(operandShape), in RangeBaseT()
184 operandShape = fn; in setOperandShapeMapping()
190 ValueShapeMapFn getOperandShapeMapping() const { return operandShape; } in getOperandShapeMapping()
224 ValueShapeMapFn operandShape;
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 377 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents() local
378 if (!operandShape.hasRank()) in inferReturnTypeComponents()
387 if (i == axis || operandShape.isDynamicDim(i)) in inferReturnTypeComponents()
390 outputShape[i] = operandShape.getDimSize(i); in inferReturnTypeComponents()
391 if (outputShape[i] != operandShape.getDimSize(i)) in inferReturnTypeComponents()
406 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents() local
410 if (!operandShape.hasRank() || operandShape.isDynamicDim(axis)) { in inferReturnTypeComponents()
415 concatDimSize += operandShape.getDimSize(axis); in inferReturnTypeComponents()
839 ShapeAdaptor operandShape, IntegerAttr axis, in ReduceInferReturnTypes() argument
841 if (!operandShape.hasRank()) { in ReduceInferReturnTypes()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Interfaces/ |
| H A D | InferTypeOpInterface.cpp | 164 if (operandShape) in getShape()
165 if (ShapeAdaptor ret = operandShape(val)) in getShape()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaInferShapes.cpp | 189 auto operandShape = [&](Value val) -> ShapeAdaptor { in propagateShapesInRegion() local
212 ValueShapeRange range(op.getOperands(), operandShape); in propagateShapesInRegion()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorUnrollDistribute.cpp | 342 SmallVector<int64_t> operandShape = applyPermutationMap( in matchAndRewrite() local
346 loc, operand, operandOffets, operandShape, operandStrides); in matchAndRewrite()
594 SmallVector<int64_t> operandShape(vecType.getShape().begin(), in matchAndRewrite() local
603 operandShape[i] = distributedDim->second; in matchAndRewrite()
606 VectorType::get(operandShape, vecType.getElementType()); in matchAndRewrite()
|