Searched refs:n_block_idx (Results 1 – 2 of 2) sorted by relevance
48 const Index n_block_idx; member892 for (Index n_block_idx = 0; n_block_idx < n_blocks; n_block_idx++) {893 const Index n_start = n_block_idx * nc;901 Index future_id = (blockAId * n_blocks + n_block_idx);908 blockBs[n_block_idx], // blockB922 n_block_idx, // n_block_idx1001 const Index set_idx = blockAId * arg.n_blocks + arg.n_block_idx;
27 const Index n_block_idx = blockIdx.y; in EigenContractionKernelInternal() local30 const Index base_n = 64 * n_block_idx; in EigenContractionKernelInternal()510 const Index n_block_idx = blockIdx.y; in EigenContractionKernel() local513 const Index base_n = 64 * n_block_idx; in EigenContractionKernel()1144 const Index n_block_idx = blockIdx.y; in EigenFloatContractionKernel() local1147 const Index base_n = 64 * n_block_idx; in EigenFloatContractionKernel()1184 const Index n_block_idx = blockIdx.y; in EigenFloatContractionKernel16x16() local1187 const Index base_n = 64 * n_block_idx; in EigenFloatContractionKernel16x16()