pytorch/torch/csrc/distributed/autograd/python_autograd.h

10 lines
174 B
C++

#pragma once
#include <torch/csrc/python_headers.h>
namespace torch::distributed::autograd {
PyMethodDef* python_functions();
} // namespace torch::distributed::autograd