mxnet
Public Attributes | List of all members
mxnet::rtc::CudaModule::ArgType Struct Reference

cuda kernel argument descriptor More...

#include <rtc.h>

Collaboration diagram for mxnet::rtc::CudaModule::ArgType:
Collaboration graph

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

Detailed Description

cuda kernel argument descriptor

Member Data Documentation

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


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