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

Use Cub-based tensor reduction in multinomial op. The Eigen implementation... 

Use Cub-based tensor reduction in multinomial op. The Eigen implementation seems to trigger a bug on Volta GPUs.

PiperOrigin-RevId: 228798930
parent 63a6b272
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