Searched defs:LaunchMaxPoolingWithArgmax (Results 1 – 1 of 1) sorted by relevance
949 struct LaunchMaxPoolingWithArgmax<CPUDevice, T, Targmax> { struct950 static void launch(OpKernelContext* context, const PoolParameters& params, in launch()1448 struct LaunchMaxPoolingWithArgmax<Eigen::GpuDevice, T, int64_t> { struct1449 static void launch(OpKernelContext* context, const PoolParameters& params, in launch()