Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cpu/
DDistanceOpsKernel.cpp138 struct idist_calc { struct
139 static inline data_t map(const data_t& diff, const data_t& p) { return diff; } in map()
140 static inline data_t red(const data_t& agg, const data_t& up) { return max(agg, up); } in red()
141 static inline scalar_t finish(const scalar_t agg, const scalar_t p) { return agg; } in finish()
144 … { return Vec(grad) * sign(diff) * (Vec(1) - vec::minimum(Vec(1), (diff.abs() - Vec(dist)).abs().c… in backward()