Searched refs:compute_coordinates (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | GridSampler.h | 142 static inline scalar_t compute_coordinates(scalar_t coord, int64_t size, in compute_coordinates() function
169 coord = compute_coordinates(coord, size, padding_mode, align_corners); in grid_sampler_compute_source_index()
224 x = compute_coordinates(x, W, padding_mode, align_corners); in get_value_bounded()
225 y = compute_coordinates(y, H, padding_mode, align_corners); in get_value_bounded()
268 x = compute_coordinates(x, W, padding_mode, align_corners); in add_value_bounded()
269 y = compute_coordinates(y, H, padding_mode, align_corners); in add_value_bounded()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | GridSampler.cuh | 150 scalar_t compute_coordinates(scalar_t coord, int size, in compute_coordinates() function
180 coord = compute_coordinates(coord, size, padding_mode, align_corners); in grid_sampler_compute_source_index()
235 x = compute_coordinates(x, W, padding_mode, align_corners); in get_value_bounded()
236 y = compute_coordinates(y, H, padding_mode, align_corners); in get_value_bounded()
289 x = compute_coordinates(x, W, padding_mode, align_corners); in add_value_bounded()
290 y = compute_coordinates(y, H, padding_mode, align_corners); in add_value_bounded()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | GridSamplerKernel.cpp | 363 inline Vec compute_coordinates(const Vec &in) const { in compute_coordinates() function
387 inline Vec compute_coordinates(const Vec &in) const { in compute_coordinates() function
416 inline Vec compute_coordinates(const Vec &in) const { in compute_coordinates() function
869 auto ix = convert_to_int_of_same_size(compute_W.compute_coordinates(x)); in get_value_bounded()
870 auto iy = convert_to_int_of_same_size(compute_H.compute_coordinates(y)); in get_value_bounded()
885 auto ix = convert_to_int_of_same_size(compute_W.compute_coordinates(x)); in add_value_bounded()
886 auto iy = convert_to_int_of_same_size(compute_H.compute_coordinates(y)); in add_value_bounded()
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 4121 def compute_coordinates(coords: Tensor, size: int) -> Tensor: function
4135 return compute_coordinates(coords_un, size)
4223 x = compute_coordinates(ix, iW)
4224 y = compute_coordinates(iy, iH)
|