Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistanceOpsKernel.cpp119 struct tdist_calc { struct
121 static inline data_t map(const data_t& diff, const data_t& p) { return diff * diff; } in map()
122 static inline data_t red(const data_t& agg, const data_t& up) { return agg + up; } in red()
123 static inline scalar_t finish(const scalar_t agg, const scalar_t p) { return std::sqrt(agg); } in finish()
124 … const scalar_t dist, const Vec& p) { return dist == 0.0 ? Vec(0) : Vec(grad) * diff / Vec(dist); } in backward()