Adding all project files
This commit is contained in:
parent
6c9e127bdc
commit
cd4316ad0f
42289 changed files with 8009643 additions and 0 deletions
|
@ -0,0 +1,21 @@
|
|||
#pragma once
|
||||
|
||||
#include <c10/core/SafePyObject.h>
|
||||
#include <c10/macros/Macros.h>
|
||||
#include <unordered_map>
|
||||
|
||||
namespace at::impl {
|
||||
|
||||
struct TORCH_API ThreadLocalPythonObjects {
|
||||
static void set(const std::string& key, std::shared_ptr<SafePyObject> value);
|
||||
static const std::shared_ptr<SafePyObject>& get(const std::string& key);
|
||||
static bool contains(const std::string& key);
|
||||
|
||||
static const ThreadLocalPythonObjects& get_state();
|
||||
static void set_state(ThreadLocalPythonObjects state);
|
||||
|
||||
private:
|
||||
std::unordered_map<std::string, std::shared_ptr<c10::SafePyObject>> obj_dict_;
|
||||
};
|
||||
|
||||
} // namespace at::impl
|
Loading…
Add table
Add a link
Reference in a new issue