1 | #pragma once |
---|---|
2 | |
3 | #include <ATen/core/DeprecatedTypeProperties.h> |
4 | #include <c10/core/TensorOptions.h> |
5 | #include <utility> |
6 | #include <vector> |
7 | |
8 | namespace torch { |
9 | namespace utils { |
10 | |
11 | std::string options_to_string(const at::TensorOptions options); |
12 | std::string type_to_string(const at::DeprecatedTypeProperties& type); |
13 | at::TensorOptions options_from_string(const std::string& str); |
14 | |
15 | // return a vector of all "declared" types, even those that weren't compiled |
16 | std::vector<std::pair<at::Backend, at::ScalarType>> all_declared_types(); |
17 | |
18 | } // namespace utils |
19 | } // namespace torch |
20 |