Caffe2 - C++ API
A deep learning, cross platform ML framework
context_base.cpp
1 #include <ATen/core/context_base.h>
2 
3 #include <c10/util/Logging.h>
4 
5 namespace at {
6 
7 C10_DEFINE_TYPED_REGISTRY(
8  ContextRegistry,
9  at::DeviceType,
10  at::BaseContext,
11  std::unique_ptr,
12  at::Device);
13 
14 } // namespace at
15 
16 namespace caffe2 {
17 
18 // TODO: rename context.h -> context_cpu.h & context_base.h -> context.h
19 
20 } // namespace caffe2
at::BaseContext
Virtual interface for the Context class in Caffe2.
Definition: context_base.h:32
c10::Device
Represents a a compute device on which a tensor is located.
Definition: Device.h:30
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13
at
Flush-To-Zero and Denormals-Are-Zero mode.
Definition: AccumulateType.h:17
Generated on Thu Mar 21 2019 13:06:08 for Caffe2 - C++ API by   doxygen 1.8.11