Searched refs:softmax_and_grad_batch_ptr_copy (Results 1 – 1 of 1) sorted by relevance
659 GpuDeviceArrayOnHost<int> softmax_and_grad_batch_ptr_copy( in CSRSparseMatrixSoftmaxGradGPUImpl() local661 TF_RETURN_IF_ERROR(softmax_and_grad_batch_ptr_copy.Init()); in CSRSparseMatrixSoftmaxGradGPUImpl()663 softmax_and_grad_batch_ptr_copy.Set(i, softmax_host_batch_ptr(i)); in CSRSparseMatrixSoftmaxGradGPUImpl()664 softmax_and_grad_batch_ptr_copy.Set(batch_size + 1 + i, in CSRSparseMatrixSoftmaxGradGPUImpl()667 TF_RETURN_IF_ERROR(softmax_and_grad_batch_ptr_copy.Finalize()); in CSRSparseMatrixSoftmaxGradGPUImpl()676 softmax_and_grad_batch_ptr_copy.data(), softmax_row_ptr.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()