|
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
1.8.11