mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
1dad051b05
Move workspace related functions to separate file Test Plan: Existing tests Differential Revision: D40657708 Pull Request resolved: https://github.com/pytorch/pytorch/pull/87651 Approved by: https://github.com/malfet
16 lines
568 B
C++
16 lines
568 B
C++
namespace caffe2 {
|
|
namespace python {
|
|
|
|
Workspace* GetCurrentWorkspace();
|
|
void SetCurrentWorkspace(Workspace* workspace);
|
|
Workspace* NewWorkspace();
|
|
Workspace* GetWorkspaceByName(const std::string& name);
|
|
std::string GetCurrentWorkspaceName();
|
|
void InsertWorkspace(const std::string& name, std::unique_ptr<Workspace> ws);
|
|
void SwitchWorkspaceInternal(const std::string& name, bool create_if_missing);
|
|
void ResetWorkspace(Workspace* workspace);
|
|
void GetWorkspaceNames(std::vector<std::string>& names);
|
|
void ClearWorkspaces();
|
|
} // namespace python
|
|
} // namespace caffe2
|