Searched defs:pooled_height (Results 1 – 7 of 7) 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()
113 const int width, const int channels, const int pooled_height, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
146 const int width, const int channels, const int pooled_height, in MaxPoolForwardNHWC()
232 const dtype* __restrict__ output_data, const int pooled_height, in MaxPoolGradBackwardNoMaskNCHW()
273 const dtype* __restrict__ output_data, const int pooled_height, in MaxPoolGradBackwardNoMaskNHWC()
356 const int width, int channels, const int pooled_height, 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()
139 const int batch, const int pooled_plane, const int pooled_height, in operator ()()
|
| D | avgpooling_op_gpu.cu.cc | 84 const int channels, const int pooled_height, in RunAvePoolBackwardNHWC()
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | PoolingLayerNode.cpp | 51 unsigned int pooled_height = 0; in compute_output_descriptor() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | AveragePool2d.cu | 35 const int64_t height, const int64_t width, const int64_t pooled_height, in avg_pool2d_out_cuda_frame()
84 const int64_t height, const int64_t width, const int pooled_height, in avg_pool2d_out_cuda_frame_nhwc()
133 const int64_t width, const int64_t pooled_height, const int64_t pooled_width, in avg_pool2d_backward_out_cuda_frame()
190 const int64_t width, const int pooled_height, const int pooled_width, in avg_pool2d_backward_out_cuda_frame_nhwc()
|
| D | DilatedMaxPool2d.cu | 47 const int64_t width, const int pooled_height, const int pooled_width, in max_pool_forward_nchw()
179 const int64_t height, const int64_t width, const int pooled_height, in max_pool_backward_nchw()
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | Types.h | 1420 unsigned int pooled_height() const in pooled_height() function
|