#pragma once

namespace torch {
namespace autograd {

void initDistFunctions(PyObject* module);

} // namespace autograd
} // namespace torch