Raman Sarokin 115623e2fc Making Softmax in OpenCL in 3 passes.
Improves numerical stability.

PiperOrigin-RevId: 347941516
Change-Id: Ibe344c9922e1e267501f42ce1123ec943ee3eb97
2020-12-16 19:26:01 -08:00
..
2020-11-23 17:43:44 -08:00
2020-11-23 17:43:44 -08:00