Home / Class/ device_type Class — pytorch Architecture

device_type Class — pytorch Architecture

Architecture documentation for the device_type class in DispatchStub.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/DispatchStub.h lines 249–253

  template <typename... ArgTypes>
  rT operator()(c10::DeviceType device_type, ArgTypes&&... args) {
    FnPtr call_ptr = get_call_ptr(device_type);
    return (*call_ptr)(std::forward<ArgTypes>(args)...);
  }

Source

Analyze Your Own Codebase

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

Try Supermodel Free