Searched refs:SoftmaxOpGPU (Results 1 – 1 of 1) sorted by relevance
129 class SoftmaxOpGPU : public OpKernel { class131 explicit SoftmaxOpGPU(OpKernelConstruction* context) : OpKernel(context) { in SoftmaxOpGPU() function in tensorflow::SoftmaxOpGPU198 SoftmaxOpGPU<Eigen::half>);201 SoftmaxOpGPU<float>);204 SoftmaxOpGPU<double>);207 SoftmaxOpGPU<Eigen::half>);210 SoftmaxOpGPU<float>);