Tensor Comprehensions
|
Specializing CudaDim to differentiate between Block and Grid sizes. More...
#include <mapping_options.h>
Public Member Functions | |
Block ()=default | |
Block (const CudaDimView &view) | |
Block (const CudaDimProto &proto) | |
Block (std::initializer_list< uint64_t > il) | |
Block (std::vector< uint64_t > il) | |
Public Member Functions inherited from tc::CudaDim | |
CudaDim () | |
CudaDim (const CudaDim &cudaDim) | |
CudaDim (const CudaDimProto &proto) | |
CudaDim (const CudaDimView &view) | |
CudaDim (std::initializer_list< uint64_t > il) | |
CudaDim (std::vector< uint64_t > il) | |
CudaDim (uint64_t x, uint64_t y=CudaDimView::defaultDim, uint64_t z=CudaDimView::defaultDim) | |
Public Member Functions inherited from tc::CudaDimView | |
CudaDimView (const CudaDimView &)=default | |
Construct a view that refers to a protocol buffers message. More... | |
CudaDimView (CudaDimProto &buf) | |
size_t | size () const |
Number of values held. More... | |
std::vector< uint64_t > | extractVector () const |
Return a copy of values as std::vector. More... | |
std::array< uint64_t, 3 > | extractDefaultedArray () const |
Return a copy of values as std::array of size 3 padded with defaultDim. More... | |
ValueAccessor< uint64_t > | operator[] (size_t i) |
uint64_t | operator[] (size_t i) const |
Access the values positionally (x=0, y=1, z=2). More... | |
CudaDimView & | operator= (const CudaDimView &view) |
Assign the values from another view. More... | |
bool | operator== (const CudaDimView &view) const |
Compare the values with those from another view. More... | |
bool | operator!= (const CudaDimView &view) const |
std::string | toCommaSeparatedString () const |
Conversion to string and output operators. More... | |
Additional Inherited Members | |
Public Attributes inherited from tc::CudaDimView | |
CudaDimProto & | proto |
Static Public Attributes inherited from tc::CudaDimView | |
static const uint64_t | defaultDim = 1 |
|
default |
|
inline |
|
inline |
|
inline |
|
inline |