execute_op Class — pytorch Architecture

Source Code

aten/src/ATen/native/cpu/Loops.h lines 99–110

template <typename func_t,
    std::enable_if_t<std::is_void_v<typename function_traits<func_t>::result_type>>* = nullptr>
inline void
execute_op(char* C10_RESTRICT data[], const int64_t* strides, int64_t i, int64_t n, func_t&& op) {
  using traits = function_traits<func_t>;
  for (; i < n; i++) {
    std::apply(op, dereference<traits>(
        &data[0],
        &strides[0],
        i));
  }
}