Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::cuda::Array< T, size > Struct Template Reference

Public Member Functions

C10_HOST_DEVICE T operator[] (int i) const
 
C10_HOST_DEVICE Toperator[] (int i)
 
C10_HIP_HOST_DEVICE Array (const Array &)=default
 
C10_HIP_HOST_DEVICE Arrayoperator= (const Array &)=default
 
C10_HOST_DEVICE Array (T x)
 

Data Fields

T data [size]
 

Detailed Description

template<typename T, int size>
struct at::cuda::Array< T, size >

Definition at line 10 of file Array.h.


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