is_nan_vec Class — pytorch Architecture

Architecture documentation for the is_nan_vec class in MaxPoolKernel.cpp from the pytorch codebase.

Class cpp

Source Code

aten/src/ATen/native/cpu/MaxPoolKernel.cpp lines 28–31

template <typename scalar_t>
vec::Vectorized<scalar_t> is_nan_vec(vec::Vectorized<scalar_t> vec) {
  return vec.isnan();
}

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