mxnet
|
cuda kernel argument descriptor More...
#include <rtc.h>
Public Attributes | |
bool | is_ndarray |
whether argument is NDArray More... | |
bool | is_const |
whether argument is constant (input) More... | |
mshadow::TypeFlag | dtype |
data type of argument More... | |
cuda kernel argument descriptor
mshadow::TypeFlag mxnet::rtc::CudaModule::ArgType::dtype |
data type of argument
bool mxnet::rtc::CudaModule::ArgType::is_const |
whether argument is constant (input)
bool mxnet::rtc::CudaModule::ArgType::is_ndarray |
whether argument is NDArray