#pragma once

namespace torch::autograd {

void initNNFunctions(PyObject* module);

}