Home
last modified time | relevance | path

Searched defs:twice_low (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.h88 static inline scalar_t reflect_coordinates(scalar_t in, int64_t twice_low, in reflect_coordinates()
111 static inline scalar_t reflect_coordinates_set_grad(scalar_t in, int64_t twice_low, in reflect_coordinates_set_grad()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cuh86 scalar_t reflect_coordinates(scalar_t in, int twice_low, int twice_high) { in reflect_coordinates()
109 scalar_t reflect_coordinates_set_grad(scalar_t in, int twice_low, int twice_high, in reflect_coordinates_set_grad()