Home
last modified time | relevance | path

Searched defs:compute_coordinates (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DGridSamplerKernel.cpp363 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.h142 static inline scalar_t compute_coordinates(scalar_t coord, int64_t size, in compute_coordinates() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cuh150 scalar_t compute_coordinates(scalar_t coord, int size, in compute_coordinates() function
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py4121 def compute_coordinates(coords: Tensor, size: int) -> Tensor: function