Llama-3.1-8B-DALv0.1
/
venv
/lib
/python3.12
/site-packages
/torchgen
/packaged
/ATen
/templates
/DispatchKeyFunction.h
// ${generated_comment} | |
// NB: The implementing C++ file is RegisterDispatchKey.cpp | |
// The only #includes we need are for custom classes that have defaults in the C++ API | |
// Forward declarations of any types needed in the operator signatures. | |
// We can't directly include these classes because it will cause circular include dependencies. | |
// This file is included by TensorBody.h, which defines the Tensor class. | |
namespace at { | |
namespace ${dispatch_namespace} { | |
${dispatch_namespaced_declarations} | |
} // namespace ${dispatch_namespace} | |
} // namespace at | |