pytorch/torch/csrc/PtrWrapper.h

18 lines
369 B
C

#ifndef THP_PTR_WRAPPER_H
#define THP_PTR_WRAPPER_H
#include "torch/csrc/python_headers.h"
#include <functional>
/**
* Python wrapper around arbitrary opaque C++ class
*/
bool THPWrapper_init(PyObject *module);
PyObject * THPWrapper_New(void *data, void (*destructor)(void*));
void * THPWrapper_get(PyObject * obj);
bool THPWrapper_check(PyObject * obj);
#endif