Searched refs:LaunchPoolingOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.h | 28 struct LaunchPoolingOp;
|
| D | pooling_ops_3d.cc | 101 struct LaunchPoolingOp<CPUDevice, T, AVG> { struct
116 struct LaunchPoolingOp<CPUDevice, T, MAX> { struct
194 LaunchPoolingOp<Device, T, Type>::launch(context, tensor_in, window, stride, in Compute()
744 struct LaunchPoolingOp<GPUDevice, T, AVG> { struct
757 struct LaunchPoolingOp<GPUDevice, T, MAX> { struct
|
| D | pooling_ops_3d_sycl.h | 178 struct LaunchPoolingOp<SYCLDevice, T, MAX> {
584 struct LaunchPoolingOp<SYCLDevice, T, AVG> {
|