Searched refs:softmax_host_batch_ptr (Results 1 – 1 of 1) sorted by relevance
625 auto softmax_host_batch_ptr = softmax.batch_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local651 DCHECK_EQ(batch_size, softmax_host_batch_ptr.size() - 1); in CSRSparseMatrixSoftmaxGradGPUImpl()660 ctx, 2 * softmax_host_batch_ptr.size()); in CSRSparseMatrixSoftmaxGradGPUImpl()662 for (int i = 0; i < softmax_host_batch_ptr.size(); ++i) { in CSRSparseMatrixSoftmaxGradGPUImpl()663 softmax_and_grad_batch_ptr_copy.Set(i, softmax_host_batch_ptr(i)); in CSRSparseMatrixSoftmaxGradGPUImpl()