Searched defs:shared_memory_size (Results 1 – 3 of 3) sorted by relevance
248 const size_t shared_memory_size = sizeof(int) * (batch_size + 1); in operator ()() local316 const size_t shared_memory_size = in CSRSparseMatrixBatchMulVecImpl() local454 const size_t shared_memory_size = sizeof(int) * (batch_size + 1); in CSRSparseMatrixSoftmaxGPUImpl() local659 const size_t shared_memory_size = 2 * sizeof(int) * (batch_size + 1); in CSRSparseMatrixSoftmaxGradGPUImpl() local
385 constexpr auto shared_memory_size = 48 << 10; // 48 KB in LaunchTopKKernel() local412 auto shared_memory_size = (num_shards + 1) * k * sizeof(Entry<T>); variable
229 int32 shared_memory_size = 0; in compute() local