Searched refs:MatMulOperandRole (Results 1 – 2 of 2) sorted by relevance
28 bool isAccumulatorOrResult(MatMulOperandRole operandType) { in isAccumulatorOrResult()29 return operandType == MatMulOperandRole::C; in isAccumulatorOrResult()70 info.operandRole = MatMulOperandRole::C; in getWarpMatrixInfo()76 info.operandRole = MatMulOperandRole::A; in getWarpMatrixInfo()80 info.operandRole = MatMulOperandRole::B; in getWarpMatrixInfo()213 if (type.operandRole == MatMulOperandRole::A || in getLdMatrixParams()214 type.operandRole == MatMulOperandRole::C) { in getLdMatrixParams()
27 enum class MatMulOperandRole : int32_t { A = 0, B, C }; enum33 MatMulOperandRole operandRole;