Commit a76a7e85 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Use CUB-based row reduction for sparse_xent_op. The Eigen implementation seems... 

Use CUB-based row reduction for sparse_xent_op. The Eigen implementation seems to trigger a bug on Volta GPUs.

PiperOrigin-RevId: 228756250
parent 02630688
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment