サポートされるネットワーク、層、クラス
GPU Coder™ は、系列または有向非循環グラフ (DAG) ニューラル ネットワークのコード生成をサポートしています。このページは、サポートされている深層学習ネットワーク、層、クラスの一覧を提供します。
コード ジェネレーターでは、ロケール設定によって決まる 8 ビット ASCII コードセットで文字が表されることに注意してください。したがって、クラス名、層の名前、層の説明、またはネットワーク名に非 ASCII 文字を使用すると、エラーが発生する可能性があります。詳細については、コード生成での文字のエンコードを参照してください。
サポートされている事前学習済みのネットワーク
Deep Learning Toolbox™ を使用するか、以下の表に示されているいずれかの事前学習済みのネットワークを使用して、1 つの CPU、1 つの GPU、または複数の GPU 上でニューラル ネットワークに学習させ、CUDA® コードを生成することができます。
以下のターゲット深層学習ライブラリについて、以下の事前学習済みネットワークが GPU Coder によるコード生成でサポートされています。
- サポートには制限がある場合があります。エラーや予期しない動作が発生する可能性があります。詳細については、リファレンス ページの「拡張機能」のセクションを参照してください。
ネットワークの名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
Caffe ネットワーク
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
サポートされている層
以下のターゲット深層学習ライブラリについて、以下の層が GPU Coder によるコード生成でサポートされています。
入力層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
|
畳み込み層と全結合層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
|
シーケンス層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
活性化層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
正規化層、ドロップアウト層、およびトリミング層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
|
プーリング層と逆プーリング層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
結合層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
|
トランスフォーマー層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
|
オブジェクト検出層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
出力層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
カスタム Keras 層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
カスタム ONNX 層
層の名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
カスタム層
サポートされているクラス
以下のターゲット深層学習ライブラリについて、以下のクラスが GPU Coder によるコード生成でサポートされています。
名前 | CUDA | TensorRT | cuDNN |
---|---|---|---|
| |||
| |||
| |||
| |||
| |||
| |||
| |||
| |||
|
参考
関数
オブジェクト
coder.gpuConfig
|coder.CodeConfig
|coder.EmbeddedCodeConfig
|coder.gpuEnvConfig
|coder.CuDNNConfig
|coder.TensorRTConfig
関連するトピック
- 事前学習済みの深層ニューラル ネットワーク (Deep Learning Toolbox)
- 転移学習入門 (Deep Learning Toolbox)
- 分類用のシンプルな深層学習ニューラル ネットワークの作成 (Deep Learning Toolbox)
- コード生成用の事前学習済みのネットワークの読み込み
- cuDNN を使用した深層学習ネットワークのコード生成
- TensorRT を使用した深層学習ネットワークのコード生成
- ARM Mali GPU をターゲットとする深層学習ネットワークのコード生成