クラス
NcclAllReduce <T extends TNumber > | すべての入力テンソルにわたるリダクションを含むテンソルを出力します。 |
NcclBroadcast <T extends TNumber > | 出力に接続されているすべてのデバイスに「入力」を送信します。 |
NcclReduce <T extends TNumber > | 「reduction」を使用して「num_devices」から「input」を単一のデバイスに減らします。 |
NcclAllReduce <T extends TNumber > | すべての入力テンソルにわたるリダクションを含むテンソルを出力します。 |
NcclBroadcast <T extends TNumber > | 出力に接続されているすべてのデバイスに「入力」を送信します。 |
NcclReduce <T extends TNumber > | 「reduction」を使用して「num_devices」から「input」を単一のデバイスに減らします。 |
特に記載のない限り、このページのコンテンツはクリエイティブ・コモンズの表示 4.0 ライセンスにより使用許諾されます。コードサンプルは Apache 2.0 ライセンスにより使用許諾されます。詳しくは、Google Developers サイトのポリシーをご覧ください。Java は Oracle および関連会社の登録商標です。
最終更新日 2024-11-08 UTC。