Caffe2 - C++ API
A deep learning, cross platform ML framework
Types.h
1 #pragma once
2 
3 #include <ATen/cudnn/cudnn-wrapper.h>
4 #include <ATen/Tensor.h>
5 
6 namespace at { namespace native {
7 
8 cudnnDataType_t getCudnnDataType(const at::Tensor& tensor);
9 
10 int64_t cudnn_version();
11 
12 }} // namespace at::cudnn
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