Searched defs:pooled_width (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 70 const int pooled_height, const int pooled_width, const int kernel_h, in MaxPoolForwardNCHW()
114 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
147 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNHWC()
233 const int pooled_width, const int channels, const int height, in MaxPoolGradBackwardNoMaskNCHW()
274 const int pooled_width, const int channels, const int height, in MaxPoolGradBackwardNoMaskNHWC()
357 const int pooled_width, const int kernel_h, const int kernel_w, in operator ()()
376 const int channels, const int pooled_height, const int pooled_width, in operator ()()
426 const int batch, const int pooled_height, const int pooled_width, in operator ()()
|
| D | pooling_ops_3d_gpu.cu.cc | 33 const int pooled_height, const int pooled_width, const int channels, in MaxPoolGradBackwardNoMaskNCDHW()
84 const int pooled_height, const int pooled_width, const int channels, in MaxPoolGradBackwardNoMaskNDHWC()
140 const int pooled_width, const int channels, const int plane, in operator ()()
|
| D | avgpooling_op_gpu.cu.cc | 85 const int pooled_width, const int kernel_h, in RunAvePoolBackwardNHWC()
|