Home
last modified time | relevance | path

Searched refs:pooled_width (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op_gpu.cu.cc70 const int pooled_height, const int pooled_width, const int kernel_h, in MaxPoolForwardNCHW() argument
75 int pw = index % pooled_width; in MaxPoolForwardNCHW()
76 int ph = (index / pooled_width) % pooled_height; in MaxPoolForwardNCHW()
77 int c = (index / pooled_width / pooled_height) % channels; in MaxPoolForwardNCHW()
78 int n = index / pooled_width / pooled_height / channels; in MaxPoolForwardNCHW()
114 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
120 int pw = index % pooled_width; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
121 int ph = (index / pooled_width) % pooled_height; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
122 int c = (index / pooled_width / pooled_height) % channels; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
123 int n = index / pooled_width / pooled_height / channels; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
[all …]
H A Dpooling_ops_3d_gpu.cu.cc33 const int pooled_height, const int pooled_width, const int channels, in MaxPoolGradBackwardNoMaskNCDHW() argument
41 int pw = index % pooled_width; in MaxPoolGradBackwardNoMaskNCDHW()
42 int ph = (index / pooled_width) % pooled_height; in MaxPoolGradBackwardNoMaskNCDHW()
43 int pp = (index / pooled_width / pooled_height) % pooled_plane; in MaxPoolGradBackwardNoMaskNCDHW()
44 int c = (index / pooled_width / pooled_height / pooled_plane) % channels; in MaxPoolGradBackwardNoMaskNCDHW()
45 int n = (index / pooled_width / pooled_height / pooled_plane / channels); in MaxPoolGradBackwardNoMaskNCDHW()
84 const int pooled_height, const int pooled_width, const int channels, in MaxPoolGradBackwardNoMaskNDHWC() argument
95 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNDHWC()
98 n /= pooled_width; in MaxPoolGradBackwardNoMaskNDHWC()
140 const int pooled_width, const int channels, const int plane, in operator ()() argument
[all …]
H A Davgpooling_op_gpu.cu.cc46 const int pooled_height, const int pooled_width, const int kernel_h, in DEFINE_GPU_KERNELS()
59 const int pwend = min(w / stride_w + 1, pooled_width); in DEFINE_GPU_KERNELS()
62 top_diff + n * pooled_height * pooled_width * channels + c; in DEFINE_GPU_KERNELS()
73 gradient += top_diff_slice[(ph * pooled_width + pw) * channels] / in DEFINE_GPU_KERNELS()
85 const int pooled_width, const int kernel_h, in RunAvePoolBackwardNHWC() argument
95 channels, pooled_height, pooled_width, kernel_h, kernel_w, stride_h, in RunAvePoolBackwardNHWC()
104 const int pooled_width, const int kernel_h, const int kernel_w,
110 const int pooled_width, const int kernel_h, const int kernel_w,
116 const int pooled_width, const int kernel_h, const int kernel_w,
H A Dmaxpooling_op_gpu.h39 const int pooled_width, const int kernel_h,
49 const int pooled_width, const int kernel_h,
75 const int pooled_height, const int pooled_width,
H A Dpooling_ops_3d_gpu.h36 const int pooled_height, const int pooled_width,
H A Davgpooling_op.h68 const int pooled_width, const int kernel_h,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAveragePool2d.cu36 const int pooled_width, const int kernel_h, const int kernel_w, in avg_pool2d_out_cuda_frame() argument
41 const int pw = index % pooled_width; in avg_pool2d_out_cuda_frame()
42 const int ph = (index / pooled_width) % pooled_height; in avg_pool2d_out_cuda_frame()
43 const int c = (index / pooled_width / pooled_height) % channels; in avg_pool2d_out_cuda_frame()
44 const int n = index / pooled_width / pooled_height / channels; 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() argument
91 const int pw = (index / channels) % pooled_width; in avg_pool2d_out_cuda_frame_nhwc()
92 const int ph = (index / channels / pooled_width) % pooled_height; in avg_pool2d_out_cuda_frame_nhwc()
93 const int n = index / channels / pooled_width / 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() argument
[all …]
H A DDilatedMaxPool2d.cu47 const int64_t width, const int pooled_height, const int pooled_width, in max_pool_forward_nchw() argument
53 int pw = index % pooled_width; in max_pool_forward_nchw()
54 int ph = (index / pooled_width) % pooled_height; in max_pool_forward_nchw()
55 int c = (index / pooled_width / pooled_height) % channels; in max_pool_forward_nchw()
56 int n = index / pooled_width / pooled_height / channels; in max_pool_forward_nchw()
86 … const int64_t width, const int pooled_height, const int pooled_width, in C10_LAUNCH_BOUNDS_1()
115 top_data = top_data + batch_id * pooled_height * pooled_width * channels; in C10_LAUNCH_BOUNDS_1()
116 top_mask = top_mask + batch_id * pooled_height * pooled_width * channels; in C10_LAUNCH_BOUNDS_1()
123 int oW = (pooled_width + gridDim.y-1) / gridDim.y; in C10_LAUNCH_BOUNDS_1()
127 int oendW = ::min(ostartW+oW, pooled_width); in C10_LAUNCH_BOUNDS_1()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEROIPoolingLayerKernel.cpp48 … ARM_COMPUTE_RETURN_ERROR_ON((pool_info.pooled_width() == 0) || (pool_info.pooled_height() == 0)); in validate_arguments()
53 …ARM_COMPUTE_RETURN_ERROR_ON((output->dimension(0) != pool_info.pooled_width()) || (output->dimensi… in validate_arguments()
129 …TensorShape output_shape(pool_info.pooled_width(), pool_info.pooled_height(), input->info()->dimen… in configure()
134 …ARM_COMPUTE_ERROR_ON((output->info()->dimension(0) != pool_info.pooled_width()) || (output->info()… in configure()
163 const int pooled_w = _pool_info.pooled_width(); in run()
H A DNEROIAlignLayerKernel.cpp114 … ARM_COMPUTE_RETURN_ERROR_ON((pool_info.pooled_width() == 0) || (pool_info.pooled_height() == 0)); in validate_arguments()
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLROIPoolingLayerKernel.cpp61 … ARM_COMPUTE_RETURN_ERROR_ON((pool_info.pooled_width() == 0) || (pool_info.pooled_height() == 0)); in validate()
66 …ARM_COMPUTE_RETURN_ERROR_ON((output->dimension(0) != pool_info.pooled_width()) || (output->dimensi… in validate()
86 …TensorShape output_shape(pool_info.pooled_width(), pool_info.pooled_height(), input->info()->dimen… in configure()
105 build_opts.add_option("-DPOOLED_DIM_X=" + support::cpp11::to_string(pool_info.pooled_width())); in configure()
H A DCLROIAlignLayerKernel.cpp53 … ARM_COMPUTE_RETURN_ERROR_ON((pool_info.pooled_width() == 0) || (pool_info.pooled_height() == 0)); in validate_arguments()
117 build_opts.add_option("-DPOOLED_DIM_X=" + support::cpp11::to_string(pool_info.pooled_width())); in configure()
/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DPoolingLayerNode.cpp50 unsigned int pooled_width = 0; in compute_output_descriptor() local
58 …std::tie(pooled_width, pooled_height) = scaled_dimensions(input_width, input_height, pool_size_x, … in compute_output_descriptor()
62 …put_descriptor.shape.set(get_dimension_idx(data_layout, DataLayoutDimension::WIDTH), pooled_width); in compute_output_descriptor()
H A DROIAlignLayerNode.cpp80 output_desc.shape.set(idx_w, _pool_info.pooled_width()); in configure_output()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DROIPoolingLayer.cpp49 …TensorShape output_shape(pool_info.pooled_width(), pool_info.pooled_height(), src.shape()[… in roi_pool_layer()
52 const int pooled_w = pool_info.pooled_width(); in roi_pool_layer()
88 for(size_t px = 0; px < pool_info.pooled_width(); ++px) in roi_pool_layer()
H A DROIAlignLayer.cpp151 …TensorShape output_shape(pool_info.pooled_width(), pool_info.pooled_height(), src.shape()[… in roi_align_layer()
155 for(size_t px = 0; px < pool_info.pooled_width(); ++px) in roi_align_layer()
172 float bin_size_x = roi_dims_x / pool_info.pooled_width(); in roi_align_layer()
/aosp_15_r20/external/pytorch/docs/cpp/source/notes/
H A Dinference_mode.rst101 int64_t pooled_width,
106 ctx->saved_data["pooled_width"] = pooled_width;
115 pooled_width, sampling_ratio, aligned);
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/roialign/generic/neon/
H A Dimpl.cpp230 const int pooled_w = pool_info.pooled_width(); in roi_align()
261 float bin_size_x = roi_dims_x / pool_info.pooled_width(); in roi_align()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/
H A DTypes.h1410 …ROIPoolingLayerInfo(unsigned int pooled_width, unsigned int pooled_height, float spatial_scale, un…
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
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DROIPoolingLayerFixture.h71 const float pool_width = pool_info.pooled_width();
H A DROIAlignLayerFixture.h72 const float pool_width = pool_info.pooled_width();
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py3982 *_, pooled_height, pooled_width = grad_output.get_size()
4024 pwend = ops.minimum(pwend, ops.index_expr(pooled_width, torch.int32))
4733 *_, pooled_height, pooled_width = grad_output.get_size()
4804 pwend = ops.minimum(pwend, ops.index_expr(pooled_width, torch.int32))
4834 pooled_width,
4904 *_, pooled_depth, pooled_height, pooled_width = grad_output.get_size()
4988 [pdend, phend, pwend], [pooled_depth, pooled_height, pooled_width]
5033 pooled_width,
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h874 output_shape.set(idx_width, pool_info.pooled_width()); in compute_roi_align_shape()
/aosp_15_r20/external/ComputeLibrary/utils/
H A DTypePrinter.h360 …os << pool_info.pooled_width() << "x" << pool_info.pooled_height() << "~" << pool_info.spatial_sca…
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DFunctionHelpers.h1856 << " ROIPooling width: " << pool_info.pooled_width() in create_roi_align_layer()