Searched defs:LaunchPoolingOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 101 struct LaunchPoolingOp<CPUDevice, T, AVG> { struct
102 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
116 struct LaunchPoolingOp<CPUDevice, T, MAX> { struct
117 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
744 struct LaunchPoolingOp<GPUDevice, T, AVG> { struct
745 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
757 struct LaunchPoolingOp<GPUDevice, T, MAX> { struct
758 static void launch(OpKernelContext* context, const Tensor& tensor_in, in launch()
|