Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
DGridSampler.h88 static inline scalar_t reflect_coordinates(scalar_t in, int64_t twice_low, in reflect_coordinates() function
/external/pytorch/aten/src/ATen/native/cuda/
DGridSampler.cuh86 scalar_t reflect_coordinates(scalar_t in, int twice_low, int twice_high) { in reflect_coordinates() function
/external/pytorch/aten/src/ATen/native/cpu/
DGridSamplerKernel.cpp231 inline Vec reflect_coordinates(const Vec &in) const { in reflect_coordinates() function
313 inline Vec reflect_coordinates(const Vec &in) const { in reflect_coordinates() function
/external/pytorch/torch/_decomp/
Ddecompositions.py4109 def reflect_coordinates(coords: Tensor, twice_low: int, twice_high: int) -> Tensor: function