/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/autograd
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/autograd/python_nn_functions.h (127B)
#pragma once
namespace torch { namespace autograd {
void initNNFunctions(PyObject* module);
}} // namespace torch::autograd