Home / Class/ is_mixed_type Class — pytorch Architecture

is_mixed_type Class — pytorch Architecture

Architecture documentation for the is_mixed_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 16–21

template <typename... Args>
inline bool is_mixed_type(const Tensor& input, const Args&... parameters) {
  const auto parameter_type = first_type(parameters...);
  return ((parameter_type != ScalarType::Undefined) &&
          (parameter_type != input.scalar_type()));
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free