Searched defs:pooled_width (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 69 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNCHW()
111 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
143 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNHWC()
182 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolBackwardNoMaskNHWC()
268 const int pooled_height, const int pooled_width, const int channels, in MaxPoolGradBackwardNoMaskNCHW()
308 const int pooled_height, const int pooled_width, const int channels, in MaxPoolGradBackwardNoMaskNHWC()
388 const int pooled_width, const int kernel_h, const int kernel_w, in operator ()()
405 const int channels, const int pooled_height, const int pooled_width, in operator ()()
434 const int channels, const int pooled_height, const int pooled_width, in operator ()()
475 const int batch, const int pooled_height, const int pooled_width, in operator ()()
|
| D | pooling_ops_3d_gpu.cu.cc | 32 const int pooled_plane, const int pooled_height, const int pooled_width, in MaxPoolGradBackwardNoMaskNCDHW()
82 const int pooled_plane, const int pooled_height, const int pooled_width, in MaxPoolGradBackwardNoMaskNDHWC()
138 const int pooled_width, const int channels, const int plane, in operator ()()
|
| D | avgpooling_op_gpu.cu.cc | 87 const int pooled_width, const int kernel_h, in RunAvePoolBackwardNHWC()
|