Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DGridSamplerKernel.cpp187 struct ComputeLocationBase<scalar_t, /*align_corners=*/true> { struct
188 using Vec = Vectorized<scalar_t>;
191 const scalar_t max_val;
193 const scalar_t scaling_factor;
195 const scalar_t low; // only used when align_corners=False
196 const scalar_t twice_span;
198 const bool empty;
200 ComputeLocationBase(int64_t size) in ComputeLocationBase() function
207 inline Vec unnormalize(const Vec &in) const { in unnormalize()
211 inline Vec clip_coordinates(const Vec &in) const { in clip_coordinates()
[all …]