xref: /aosp_15_r20/external/pytorch/torch/csrc/utils/python_symnode.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/csrc/utils/python_symnode.h>
2 
3 namespace torch {
4 
get_symint_class()5 py::handle get_symint_class() {
6   // NB: leak
7   static py::handle symint_class =
8       py::object(py::module::import("torch").attr("SymInt")).release();
9   return symint_class;
10 }
11 
get_symfloat_class()12 py::handle get_symfloat_class() {
13   // NB: leak
14   static py::handle symfloat_class =
15       py::object(py::module::import("torch").attr("SymFloat")).release();
16   return symfloat_class;
17 }
18 
get_symbool_class()19 py::handle get_symbool_class() {
20   // NB: leak
21   static py::handle symbool_class =
22       py::object(py::module::import("torch").attr("SymBool")).release();
23   return symbool_class;
24 }
25 
26 } // namespace torch
27