1 #pragma once 2 3 #include <ATen/cpu/vec/functional_base.h> 4 #include <ATen/cpu/vec/functional_bfloat16.h> 5