1 | #pragma once |
---|---|
2 | |
3 | #include <c10/core/ScalarType.h> |
4 | #include <c10/macros/Macros.h> |
5 | |
6 | namespace caffe2 { |
7 | class TypeMeta; |
8 | } // namespace caffe2 |
9 | |
10 | namespace c10 { |
11 | C10_API void set_default_dtype(caffe2::TypeMeta dtype); |
12 | C10_API const caffe2::TypeMeta get_default_dtype(); |
13 | C10_API ScalarType get_default_dtype_as_scalartype(); |
14 | C10_API const caffe2::TypeMeta get_default_complex_dtype(); |
15 | } // namespace c10 |
16 |