Searched defs:ComputeLocation (Results 1 – 2 of 2) sorted by relevance
351 struct ComputeLocation<scalar_t, GridSamplerPadding::Zeros, align_corners> struct353 using Vec = Vectorized<scalar_t>;359 inline Vec apply(const Vec &in) const { in apply()363 inline Vec compute_coordinates(const Vec &in) const { in compute_coordinates()367 inline std::pair<Vec, Vec> apply_get_grad(const Vec &in) const { in apply_get_grad()373 struct ComputeLocation<scalar_t, GridSamplerPadding::Border, align_corners> struct375 using Vec = Vectorized<scalar_t>;383 inline Vec apply(const Vec &in) const { in apply()387 inline Vec compute_coordinates(const Vec &in) const { in compute_coordinates()391 inline std::pair<Vec, Vec> apply_get_grad(const Vec &in) const { in apply_get_grad()[all …]
3620 "ComputeLocation":{ object