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

Public Member Functions

 LayoutWrapper (const TensorCPU &tensor)
 
 LayoutWrapper (const dnnPrimitive_t primitive, const dnnResourceType_t type)
 
 LayoutWrapper (const size_t dimension, const size_t size[], const size_t strides[])
 
void Reset (const TensorCPU &tensor)
 
void Reset (const dnnPrimitive_t primitive, const dnnResourceType_t type)
 
void Reset (const size_t dimension, const size_t size[], const size_t strides[])
 
void Reset ()
 
 operator dnnLayout_t () const
 

Detailed Description

template<typename T>
class caffe2::mkl::LayoutWrapper< T >

Definition at line 65 of file mkl_memory.h.


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