Searched defs: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::SoftmaxOpGPU