xref: /aosp_15_r20/external/pytorch/tools/autograd/templates/python_functions.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #pragma once
2 
3 #include <Python.h>
4 
5 // ${generated_comment}
6 
7 // Python bindings for automatically generated autograd functions
8 
9 namespace torch { namespace autograd { namespace generated {
10 
11 ${shard_forward_declare}
12 
initialize_autogenerated_functions(PyObject * module)13 inline void initialize_autogenerated_functions(PyObject* module) {
14   ${shard_call}
15 }
16 
17 }}} // namespace torch::autograd::generated
18