Mila::Dnn::Compute::Cuda::Softmax::Detail Namespace Reference

Namespace for CUDA softmax implementation details. More...

Classes
struct	cuda_softmax_impl
struct	cuda_softmax_impl< float >
struct	cuda_softmax_impl< half >

Detailed Description

Namespace for CUDA softmax implementation details.

Contains specialized templates for different data types (float, half) and kernel dispatch logic.