Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistanceOpsKernel.cpp129 struct pdist_calc { struct
130 static inline data_t map(const data_t& diff, const data_t& p) { return pow(diff, p); } in map()
131 static inline data_t red(const data_t& agg, const data_t& up) { return agg + up; } in red()
132 …tic inline scalar_t finish(const scalar_t agg, const scalar_t p) { return std::pow(agg, 1.0 / p); } in finish()
133 … { return dist == 0.0 ? Vec(0) : diff * diff.abs().pow(p - Vec(2)) * Vec(grad) / Vec(dist).pow(p -… in backward()