ReduceMultiply Class — pytorch Architecture

Source Code

aten/src/ATen/native/cpu/ScatterGatherKernel.cpp lines 32–43

class ReduceMultiply {
public:
  template <typename scalar_t>
  constexpr void operator() (at::opmath_type<scalar_t> * self_data, scalar_t * src_data) const {
    using opmath_t = at::opmath_type<scalar_t>;
    *self_data *= opmath_t(c10::load(src_data));
  }

  constexpr void operator() (bool * self_data, bool * src_data) const {
    *self_data = c10::load(self_data) && c10::load(src_data);
  }
};