サポートされるネットワーク、層、クラス
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 をターゲットとする深層学習ネットワークのコード生成