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