Searched defs:softmax_traits (Results 1 – 1 of 1) sorted by relevance
62 struct softmax_traits { struct63 using accumulator_type = T;67 struct softmax_traits<Eigen::half> { struct68 using accumulator_type = float;