set_result Class — pytorch Architecture

Source Code

aten/src/ATen/native/cpu/Reduce.h lines 115–122

template<typename traits, typename res_t>
static void set_result(const int index, const res_t result, const TensorIteratorBase &iter, const int num_outputs) {
  // static_assert(std::is_same_v<res_t, typename traits::arg2_t>, "data types must match");
  if (index < num_outputs) {
    char *out = (char *) iter.data_ptr(index);
    *(res_t *) out = result;
  }
}