Skip to content

Create a network definition from scratch by TensorRT #1

@daohu527

Description

@daohu527

Reference topic create-network-def-scratch

The advantage of using tensorRT directly to create a network is to avoid the problems of different onnx exports.

  • model -> onnx -> TensorRT
  • model -> TensorRT

solution

INetworkDefinition

Existing network types are added through addxx, and custom types are added through addPluginV2.

  • addConvolution, addFullyConnected ...
  • addPluginV2

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions