Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::detail::TensorDescriptors< T > Class Template Reference

Public Member Functions

 TensorDescriptors (size_t n, std::vector< int > &dim, std::vector< int > &stride)
 
const miopenTensorDescriptor_t * descs () const
 
 TensorDescriptors (size_t n, const std::vector< int > &dim, const std::vector< int > &stride)
 
const cudnnTensorDescriptor_t * descs () const
 

Detailed Description

template<typename T>
class caffe2::detail::TensorDescriptors< T >

Definition at line 14 of file recurrent_op_miopen.h.


The documentation for this class was generated from the following files: