pytorch Architecture Docs
Home / Class/ forEachSupportedScalarType Class — pytorch Architecture

forEachSupportedScalarType Class — pytorch Architecture

Architecture documentation for the forEachSupportedScalarType class in DeviceCapability.h from the pytorch codebase.

Class c

Entity Profile

Source Code

c10/core/DeviceCapability.h lines 61–71

  template <typename F>
  void forEachSupportedScalarType(F&& visitor) const {
#define VISIT_SCALAR_TYPE(_1, n)                        \
  if (capability_data.supported_scalar_types.has_##n) { \
    visitor(ScalarType::n);                             \
  }

    AT_FORALL_SCALAR_TYPES_WITH_COMPLEX_AND_QINTS(VISIT_SCALAR_TYPE)

#undef VISIT_SCALAR_TYPE
  }

Source

Analyze Your Own Codebase

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

Try Supermodel Free