Searched refs:grad_softmax_host_batch_ptr (Results 1 – 1 of 1) sorted by relevance
629 auto grad_softmax_host_batch_ptr = grad_softmax.batch_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local652 DCHECK_EQ(batch_size, grad_softmax_host_batch_ptr.size() - 1); in CSRSparseMatrixSoftmaxGradGPUImpl()665 grad_softmax_host_batch_ptr(i)); in CSRSparseMatrixSoftmaxGradGPUImpl()