Caffe2 - C++ API
A deep learning, cross platform ML framework
types.h
1 #pragma once
2 
3 #include <ATen/ATen.h>
4 
5 #include <c10/util/Optional.h>
6 
7 #include <torch/csrc/autograd/generated/variable_factories.h>
8 #include <torch/csrc/autograd/variable.h>
9 
10 namespace torch {
11 using namespace at; // NOLINT
12 
13 using c10::optional;
14 using c10::nullopt;
15 
16 using Dtype = at::ScalarType;
17 
19 constexpr auto kUInt8 = at::kByte;
20 constexpr auto kInt8 = at::kChar;
21 constexpr auto kInt16 = at::kShort;
22 constexpr auto kInt32 = at::kInt;
23 constexpr auto kInt64 = at::kLong;
24 constexpr auto kFloat16 = at::kHalf;
25 constexpr auto kFloat32 = at::kFloat;
26 constexpr auto kFloat64 = at::kDouble;
27 
29 constexpr auto kU8 = kUInt8;
30 constexpr auto kI8 = kInt8;
31 constexpr auto kI16 = kInt16;
32 constexpr auto kI32 = kInt32;
33 constexpr auto kI64 = kInt64;
34 constexpr auto kF16 = kFloat16;
35 constexpr auto kF32 = kFloat32;
36 constexpr auto kF64 = kFloat64;
37 } // namespace torch
Definition: jit_type.h:17
Flush-To-Zero and Denormals-Are-Zero mode.