Home
last modified time | relevance | path

Searched refs:access_y (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSample.cuh193 int access_y = max(min(y, height - 1), 0); in upsample_get_value_bounded() local
195 return data[batch][channel][access_y][access_x]; in upsample_get_value_bounded()
209 int access_y = max(min(y, height - 1), 0); in upsample_increment_value_bounded() local
215 &data[batch][channel][access_y][access_x], static_cast<scalar_t>(value)); in upsample_increment_value_bounded()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h375 int64_t access_y = std::max(std::min(y, height - 1), static_cast<int64_t>(0)); in upsample_get_value_bounded() local
376 return data[access_y * width + access_x]; in upsample_get_value_bounded()
388 int64_t access_y = std::max(std::min(y, height - 1), static_cast<int64_t>(0)); in upsample_increment_value_bounded() local
389 data[access_y * width + access_x] += value; in upsample_increment_value_bounded()