Searched refs:kFactor (Results 1 – 1 of 1) sorted by relevance
373 int64_t kFactor; in verify() local377 kFactor = 4; in verify()384 kFactor = 8; in verify()391 kFactor = 32; in verify()394 kFactor = 128; in verify()398 kFactor = 16; in verify()414 int64_t unitA = (mmaShape[0] / 8) * (mmaShape[2] / kFactor); in verify()415 int64_t unitB = (mmaShape[1] / 8) * (mmaShape[2] / kFactor); in verify()418 allowedShapes.push_back({16, 8, kFactor}); in verify()419 allowedShapes.push_back({16, 8, kFactor * 2}); in verify()