Searched defs:LaunchPoolingOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 96 struct LaunchPoolingOp<CPUDevice, T, AVG> { struct
97 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
111 struct LaunchPoolingOp<CPUDevice, T, MAX> { struct
112 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
778 struct LaunchPoolingOp<GPUDevice, T, AVG> { struct
779 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
791 struct LaunchPoolingOp<GPUDevice, T, MAX> { struct
792 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
|