pytorch Architecture Docs
Home / Class/ promote_type Class — pytorch Architecture

promote_type Class — pytorch Architecture

Architecture documentation for the promote_type class in autocast_mode.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/autocast_mode.h lines 301–309

template <typename Arg0, typename... Args>
inline at::ScalarType promote_type(
    at::ScalarType current,
    c10::DeviceType device_type,
    Arg0 arg0,
    Args... args) {
  auto new_current = prioritize(current, arg0, device_type);
  return promote_type(new_current, device_type, args...);
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free