is_masked Class — pytorch Architecture
Architecture documentation for the is_masked class in vec256_mask.h from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/cpu/vec/vec256/vec256_mask.h lines 232–235
template <>
inline bool VecMask<int, 1>::is_masked(int i) const {
return _mm256_movemask_ps(_mm256_castsi256_ps(mask_[0])) & (1 << i);
}
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free