2

私のネットワークには、現在の tensorRT でサポートされていない特定のレイヤーが含まれています。そのため、tensorRT で conv レイヤーとプーリング レイヤーを実行し、tensorRT からの出力を、いくつかの特定のレイヤーを含むカフェ モデルの入力として使用したいと考えています。参照できる API またはサンプル コードはありますか? ありがとう

4

2 に答える 2