Home
last modified time | relevance | path

Searched defs:softmax_traits (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsoftmax_op_gpu.cu.cc62 struct softmax_traits { struct
63 using accumulator_type = T;
67 struct softmax_traits<Eigen::half> { struct
68 using accumulator_type = float;