Searched defs:pooled_width (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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 ()()
|
| H A 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 ()()
|
| H A D | avgpooling_op_gpu.cu.cc | 85 const int pooled_width, const int kernel_h, in RunAvePoolBackwardNHWC()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool2d.cu | 36 const int pooled_width, const int kernel_h, const int kernel_w, in avg_pool2d_out_cuda_frame()
85 const int pooled_width, const int kernel_h, const int kernel_w, 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()
|
| H A D | DilatedMaxPool2d.cu | 47 const int64_t width, const int pooled_height, const int pooled_width, in max_pool_forward_nchw()
180 const int pooled_width, const int kernel_h, const int kernel_w, in max_pool_backward_nchw()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | PoolingLayerNode.cpp | 50 unsigned int pooled_width = 0; in compute_output_descriptor() local
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | Types.h | 1411 …: _pooled_width(pooled_width), _pooled_height(pooled_height), _spatial_scale(spatial_scale), _samp… in _pooled_width() argument
1415 unsigned int pooled_width() const in pooled_width() function
|