UNARY_OUTER_LOOP Class — pytorch Architecture
Architecture documentation for the UNARY_OUTER_LOOP class in Reduce.h from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/native/cpu/Reduce.h lines 70–77
template <typename F>
inline void UNARY_OUTER_LOOP(char* data[2], const int64_t strides[2], int64_t n, F f) {
for ([[maybe_unused]] const auto j : c10::irange(n)) {
f();
data[0] += strides[0];
data[1] += strides[1];
}
}
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free