Searched defs:shared_memory_size (Results 1 – 4 of 4) 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
186 int shared_memory_size() const { in shared_memory_size() function