Fields

tensorrt.FieldType

The possible field types for the custom layer.

Members:

FLOAT

INT32

CHAR

DIMS

DATATYPE

UNKNOWN

class tensorrt.FieldMap(self: tensorrt.tensorrt.FieldMap, name: str, data: capsule, type: tensorrt.tensorrt.FieldType, length: int = 1) → None

This is a class containing an array of field params used as a layer parameter for plugin layers. The node fields are passed by the parser to the API through the plugin constructor. The implementation of the plugin should parse the contents of the FieldMap as part of the plugin constructor.

Variables

• name – str field param

• data – capsule field param

• type – FieldType field param

• length – int field param

class tensorrt.FieldCollection

This class contains an array of FieldMap s.

Variables

• num_fields – int The number of FieldMap s.

• fields – capsule The array of FieldMap s.