Searched defs:blocks_per_slice (Results 1 – 1 of 1) sorted by relevance
482 uint32_t blocks_per_slice, in computeBlockwiseKthCounts()633 uint32_t blocks_per_slice; member in at::native::mbtopk::BlockIdxToKey635 BlockIdxToKey(uint32_t blocks_per_slice): blocks_per_slice(blocks_per_slice) {} in BlockIdxToKey()663 uint32_t blocks_per_slice = at::ceil_div((int64_t)inputSliceSize, (int64_t)items_per_block); in launch() local