Searched refs:src_dim_h (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest3d.cu | 48 size_t src_dim_h, in upsample_nearest3d_out_frame() argument
63 int src_c_stride = src_dim_d * src_dim_h * src_dim_w; in upsample_nearest3d_out_frame()
70 int src_y = nn_compute_source_index_fn(height_scale, dst_y, src_dim_h); in upsample_nearest3d_out_frame()
75 int src_idx = c * src_c_stride + src_z * src_dim_h * src_dim_w + in upsample_nearest3d_out_frame()
93 size_t src_dim_h, in upsample_nearest3d_backward_out_frame() argument
108 int src_c_stride = src_dim_d * src_dim_h * src_dim_w; in upsample_nearest3d_backward_out_frame()
121 int src_y = nn_bw_compute_source_index_fn(height_scale, dst_y, src_dim_h); in upsample_nearest3d_backward_out_frame()
122 int src_y_up = nn_bw_compute_source_index_fn(height_scale, dst_y+1, src_dim_h); in upsample_nearest3d_backward_out_frame()
136 z * src_dim_h * src_dim_w + y * src_dim_w + x; in upsample_nearest3d_backward_out_frame()
|
| H A D | UpSampleNearest2d.cu | 118 size_t src_dim_h, in upsample_nearest2d_backward_out_frame() argument
130 int src_c_stride = src_dim_h * src_dim_w; in upsample_nearest2d_backward_out_frame()
138 nn_bw_compute_source_index_fn(height_scale, dst_y, src_dim_h); in upsample_nearest2d_backward_out_frame()
140 height_scale, dst_y + 1, src_dim_h); in upsample_nearest2d_backward_out_frame()
|