frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Pytorch Edge] Extend runtime compatibility to custom classes (#66972)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66972

Add api to view how many custom classes we have and what their names are

Test Plan: unit test

Reviewed By: cccclai

Differential Revision: D31811337

fbshipit-source-id: 9f8ca1fc578a0a5360c9cd8f95475acc33f250e4
This commit is contained in:
Jacob Szwejbka
2021-10-25 13:38:50 -07:00
committed by Facebook GitHub Bot
parent f4dd88489a
commit a6d0339492
4 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/custom_class.h
View File

@ -441,4 +441,6 @@ inline class_<CurClass> Library::class_(const std::string& className) {
return torch::class_<CurClass>(*ns_, className);
}
}
const std::unordered_set<std::string> getAllCustomClassesNames();
} // namespace torch