/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils/python_tuples.h (684B)
#pragma once
#include
#include
#include
#include
inline void THPUtils_packInt64Array(PyObject *tuple, size_t size, const int64_t *sizes) {
for (size_t i = 0; i != size; ++i) {
PyObject *i64 = THPUtils_packInt64(sizes[i]);
if (!i64) {
throw python_error();
}
PyTuple_SET_ITEM(tuple, i, i64);
}
}
inline PyObject* THPUtils_packInt64Array(size_t size, const int64_t *sizes) {
THPObjectPtr tuple(PyTuple_New(size));
if (!tuple) throw python_error();
THPUtils_packInt64Array(tuple.get(), size, sizes);
return tuple.release();
}