Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
torch
csrc
api
include
torch
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
c10::optional
Definition:
Optional.h:118
torch
Definition:
jit_type.h:17
at
Flush-To-Zero and Denormals-Are-Zero mode.
Definition:
AccumulateType.h:17
Generated on Thu Mar 21 2019 13:06:12 for Caffe2 - C++ API by
1.8.11