set_opt_dtype Class — pytorch Architecture

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

Class c

Source Code

aten/src/ATen/autocast_mode.h lines 392–395

template <typename T>
inline T set_opt_dtype(at::ScalarType to_type, T arg) {
  return arg;
}

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