impl_func_cum_ops Class — pytorch Architecture

Architecture documentation for the impl_func_cum_ops class in ReduceOps.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/native/ReduceOps.cpp lines 493–508

template <class Stub>
static void impl_func_cum_ops(
    const Tensor& self,
    int64_t dim,
    const Tensor& result,
    Stub& stub) {
  NoNamesGuard guard;
  if (self.dim() == 0) {
    result.fill_(self);
  } else if (self.numel() == 0) {
    result.zero_();
  } else {
    dim = maybe_wrap_dim(dim, self.dim());
    stub(self.device().type(), result, self.to(result.scalar_type()), dim);
  }
}

Source

View on GitHub

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free