Searched defs:operandShape (Results 1 – 6 of 6) sorted by relevance
39 static std::array<int64_t, 2> getTileShape(ArrayRef<int64_t> operandShape, in getTileShape() argument 156 ArrayRef<int64_t> operandShape, in getRegisterIndexToTileOffsetMap() argument 176 ArrayRef<int64_t> operandShape = fragmentType.vectorType.getShape(); in getLaneIdAndValueIdToOperandCoord() local 244 ArrayRef<int64_t> operandShape = params.fragmentType.getShape(); in getLaneIdToLdMatrixMatrixCoord() local [all...]
143 SmallVector<int64_t> operandShape = in matchAndRewrite() local
406 auto operandShape = cast<ShapedType>(operands[it.index()].getType()); in castAwayContractionLeadingOneDim() local
289 SmallVector<int64_t> operandShape = applyPermutationMap( in matchAndRewrite() local
633 ShapeAdaptor operandShape(operand.getType()); inferReturnTypeComponents() local 666 ShapeAdaptor operandShape(operand.getType()); inferReturnTypeComponents() local 1246 ReduceInferReturnTypes(ShapeAdaptor operandShape,Type inputType,IntegerAttr axis,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) ReduceInferReturnTypes() argument [all...]
348 ArrayRef<int64_t> operandShape = genericOp.getShape(opOperand); dropUnitExtentFromOperandMetadata() local