/usr/local/lib64/python3.6/site-packages/torch/include/ATen/core
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/core/PythonModeTLS.h (403B)
#pragma once
#include
#include
#include
namespace at {
namespace impl {
struct TORCH_API PythonModeTLS {
static void set_state(const std::shared_ptr& state);
static const std::shared_ptr& get_state();
static void reset_state();
};
} // namespace impl
} // namespace at