Home
last modified time | relevance | path

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

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