インフィードエンキュー事前線形化バッファ

パブリック最終クラスInfeedEnqueuePrelinearizedBuffer

事前に線形化されたバッファを TPU インフィードにエンキューする操作。

ネストされたクラス

クラスInfeedEnqueuePrelinearizedBuffer.Options InfeedEnqueuePrelinearizedBufferのオプションの属性

パブリックメソッド

静的InfeedEnqueuePrelinearizedBuffer
create (スコープscope、オペランド<?>入力、オプション...オプション)
新しい InfeedEnqueuePrelinearizedBuffer オペレーションをラップするクラスを作成するファクトリ メソッド。
静的InfeedEnqueuePrelinearizedBuffer.Options
deviceOrdinal (長い deviceOrdinal)

継承されたメソッド

パブリックメソッド

public static InfeedEnqueuePrelinearizedBuffer create (スコープスコープ、オペランド<?> 入力、オプション...オプション)

新しい InfeedEnqueuePrelinearizedBuffer オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
入力線形化された出力を表すバリアント テンソル。
オプションオプションの属性値を持ちます
戻り値
  • InfeedEnqueuePrelinearizedBuffer の新しいインスタンス

public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)

パラメーター
デバイス通常使用する TPU デバイス。これは、Op が TPU デバイスで実行されている場合は -1、Op が CPU デバイスで実行されている場合は = 0 である必要があります。