| Name |
Last commit
58708021
-
Fix the stream issue with NCCL2 on CUDA 9.2 and later
|
History
|
Last Update |
|---|---|---|
| .. | ||
| CMakeLists.txt | Loading commit data... | |
| generic_loss_op_kernel.cc | Loading commit data... | |
| generic_loss_op_kernel.cu | Loading commit data... | |
| nll_loss_op_kernel.cc | Loading commit data... | |
| nll_loss_op_kernel.cu | Loading commit data... | |
| sigmoid_ce_loss_op_kernel.cc | Loading commit data... | |
| sigmoid_ce_loss_op_kernel.cu | Loading commit data... | |
| sigmoid_focal_loss_op_kernel.cc | Loading commit data... | |
| sigmoid_focal_loss_op_kernel.cu | Loading commit data... | |
| smooth_l1_loss_op_kernel.cc | Loading commit data... | |
| smooth_l1_loss_op_kernel.cu | Loading commit data... | |
| softmax_ce_loss_op_kernel.cc | Loading commit data... | |
| softmax_ce_loss_op_kernel.cu | Loading commit data... | |
| sparse_softmax_ce_loss_op_kernel.cc | Loading commit data... | |
| sparse_softmax_ce_loss_op_kernel.cu | Loading commit data... |