Home / Class/ cast Class — pytorch Architecture

cast Class — pytorch Architecture

Architecture documentation for the cast class in jit_type_base.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/core/jit_type_base.h lines 381–388

  template <typename T, std::enable_if_t<detail::IsSingletonType<T>::value, bool> = true>
  typename detail::CastReturnType<T>::type cast() {
    if (T::Kind == kind()) {
      TORCH_INTERNAL_ASSERT_DEBUG_ONLY(this == T::get().get());
      return typename detail::CastReturnType<T>::type(static_cast<T*>(this));
    }
    return nullptr;
  }

Source

Analyze Your Own Codebase

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

Try Supermodel Free