cudlaModuleAttribute

Defined in /dvs/p4/build/sw/rel/gpgpu/toolkit/r13.4/cuDLA/inc/cudla.h

union cudlaModuleAttribute

Module attribute.

Public Members

cudlaModuleTensorDescriptor *inputTensorDesc

Returns an array of input tensor descriptors.

uint32_t numInputTensors

Returns the number of input tensors.

uint32_t numOutputTensors

Returns the number of output tensors.

cudlaModuleTensorDescriptor *outputTensorDesc

Returns an array of output tensor descriptors.