Searched defs:pooled_height (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 68 const int height, const int width, const int pooled_height, in MaxPoolForwardNCHW()
110 const int width, const int channels, const int pooled_height, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
142 const int width, const int channels, const int pooled_height, in MaxPoolForwardNHWC()
181 const int width, const int channels, const int pooled_height, 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()
387 const int width, int channels, const int pooled_height, 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()
137 const int batch, const int pooled_plane, const int pooled_height, in operator ()()
|
| D | avgpooling_op_gpu.cu.cc | 86 const int channels, const int pooled_height, in RunAvePoolBackwardNHWC()
|