1 | #pragma once |
---|---|
2 | |
3 | #include <ATen/ATen.h> |
4 | #include <string> |
5 | #include <tuple> |
6 | |
7 | namespace torch { |
8 | namespace utils { |
9 | |
10 | std::pair<std::string, std::string> getDtypeNames(at::ScalarType scalarType); |
11 | |
12 | void initializeDtypes(); |
13 | |
14 | } // namespace utils |
15 | } // namespace torch |
16 |