Searched defs:SoftmaxOpGPU (Results 1 – 1 of 1) sorted by relevance
195 class SoftmaxOpGPU : public OpKernel { class197 explicit SoftmaxOpGPU(OpKernelConstruction* context) : OpKernel(context) { in SoftmaxOpGPU() function in tensorflow::SoftmaxOpGPU