/usr/local/lib64/python3.6/site-packages/torch/include/ATen/core
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/core/UnsafeFromTH.h (708B)
#pragma once
#include
namespace at {
inline Tensor unsafeTensorFromTH(void * th_pointer, bool retain) {
auto tensor_impl = c10::intrusive_ptr::reclaim(static_cast(th_pointer));
if (retain && tensor_impl.get() != UndefinedTensorImpl::singleton()) {
c10::raw::intrusive_ptr::incref(tensor_impl.get());
}
return Tensor(std::move(tensor_impl));
}
inline Storage unsafeStorageFromTH(void * th_pointer, bool retain) {
if (retain && th_pointer) {
c10::raw::intrusive_ptr::incref(static_cast(th_pointer));
}
return Storage(c10::intrusive_ptr::reclaim(static_cast(th_pointer)));
}
}