Searched defs:shared_memory_size (Results 1 – 2 of 2) sorted by relevance
261 const size_t shared_memory_size = sizeof(int) * (batch_size + 1); in operator ()() local329 const size_t shared_memory_size = in CSRSparseMatrixBatchMulVecImpl() local467 const size_t shared_memory_size = sizeof(int) * (batch_size + 1); in CSRSparseMatrixSoftmaxGPUImpl() local672 const size_t shared_memory_size = 2 * sizeof(int) * (batch_size + 1); in CSRSparseMatrixSoftmaxGradGPUImpl() local
226 int32 shared_memory_size = 0; in compute() local