Home / Class/ out_device Class — pytorch Architecture

out_device Class — pytorch Architecture

Architecture documentation for the out_device class in TensorCompare.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/native/TensorCompare.cpp lines 574–582

template <typename... Args>
static Device out_device(Args&... inps) {
  for (const auto& i : {inps...}) {
    if (i.device() != at::kCPU) {
      return i.device();
    }
  }
  return at::kCPU;
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free