The HPS plugin has plugin class,
HpsPlugin, with the registration name
The HPS plugin accepts one input.
The input data type must be
The input shape must be
The HPS plugin generates one output.
The output data type is
The output shape is
(batch_size, num_keys_per_sample, embedding_vector_size).
This plugin works for network with graph node named
HPS_TRT. This is also the plugin name that should be used when getting the
HpsPluginCreator from the Plugin Registry.