Searched defs:LaunchMaxPoolingWithArgmax (Results 1 – 1 of 1) sorted by relevance
887 struct LaunchMaxPoolingWithArgmax<CPUDevice, T> { struct888 static void launch(OpKernelContext* context, const PoolParameters& params, in launch()1318 struct LaunchMaxPoolingWithArgmax<Eigen::GpuDevice, T> { struct1319 static void launch(OpKernelContext* context, const PoolParameters& params, in launch()