check_mixed_data_type Class — pytorch Architecture

Architecture documentation for the check_mixed_data_type class in mixed_data_type.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/cpu/mixed_data_type.h lines 30–35

template <typename... Args>
inline void check_mixed_data_type(const Tensor& input, const Tensor& parameter, const Args&... parameters) {
  TORCH_CHECK(!parameter.defined() || parameter.scalar_type() == ScalarType::Float,
      "mixed dtype (CPU): expect parameter to have scalar type of Float");
  check_mixed_data_type(input, parameters...);
}

Source

View on GitHub

Analyze Your Own Codebase

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

Try Supermodel Free