Home / Class/ iter Class — pytorch Architecture

iter Class — pytorch Architecture

Architecture documentation for the iter class in DistributionTemplates.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/cpu/DistributionTemplates.h lines 57–65

template<typename RNG>
struct RandomFromToKernel {
  void operator()(TensorIteratorBase& iter, uint64_t range, int64_t base, std::optional<Generator> gen) {
    random_from_to_kernel(iter, range, base, check_generator<RNG>(gen));
  }
  void operator()(TensorIteratorBase& iter, std::optional<Generator> gen) {
    random_full_64_bits_range_kernel(iter, check_generator<RNG>(gen));
  }
};

Source

Analyze Your Own Codebase

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

Try Supermodel Free