Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistanceOpsKernel.cpp101 struct odist_calc { struct
102 static inline data_t map(const data_t& diff, const data_t& p) { return diff; } in map()
103 static inline data_t red(const data_t& agg, const data_t& up) { return agg + up; } in red()
104 static inline scalar_t finish(const scalar_t agg, const scalar_t /*p*/) { return agg; } in finish()
105 … const scalar_t grad, const scalar_t /*dist*/, const Vec& /*p*/) { return Vec(grad) * sign(diff); } in backward()