pytorch

1
#include <torch/csrc/utils/python_symnode.h>
2

3
namespace torch {
4

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

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

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