firstarg_is_eligible Class — pytorch Architecture
Architecture documentation for the firstarg_is_eligible class in autocast_mode.h from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/autocast_mode.h lines 397–403
template <typename... Args>
inline bool firstarg_is_eligible(
c10::DeviceType device_type,
const Tensor& arg,
Args... args) {
return is_eligible(arg, device_type);
}
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free