pytorch

1
#include "caffe2/python/pybind_state_registry.h"
2

3
namespace caffe2 {
4
namespace python {
5

6
namespace py = pybind11;
7

8
C10_DEFINE_REGISTRY(PybindAdditionRegistry, PybindAddition, py::module&);
9

10
} // namespace python
11
} // namespace caffe2
12