cudlaModuleAttribute

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