org.tensorflow.op.xla

クラス

BroadcastHelper <T はTTypeを拡張 >	XLA スタイルのブロードキャストを実行するためのヘルパー オペレーター 二項演算子に対する XLA のブロードキャスト ルールを使用して、'lhs' と 'rhs' のランクが低い方にサイズ 1 の次元を追加することにより、'lhs' と 'rhs' を同じランクにブロードキャストします。
ClusterOutput <T extends TType >	XLA 計算の出力を他のコンシューマー グラフ ノードに接続する演算子。
Conv <T extends TType >	XLA ConvGeneralDirated 演算子をラップします。ドキュメントは次のとおりです。 https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution 。
逆量子化	パックされた uint32 入力を受け取り、その入力を uint8 にアンパックして実行します。 デバイス上の逆量子化。
ドット<T extends TType >	XLA DotGeneral 演算子をラップします。ドキュメントは次の場所にあります。 https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral 。
DynamicSlice <T はTTypeを拡張 >	XLA DynamicSlice オペレーターをラップします。ドキュメントは次のとおりです。 https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice 。
DynamicUpdateSlice <T はTTypeを拡張 >	XLA DynamicUpdateSlice オペレーターをラップします。ドキュメントは次のとおりです。 https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice 。
Einsum <T extends TType >	2 入力 1 出力の基本的な einsum 演算をサポートする演算です。
収集<T extends TType >	XLA Gather オペレーターをラップします。 https://www.tensorflow.org/xla/operation_semantics#gather
KeyValueSort <T extends TNumber 、U extends TType >	XLA ソート演算子をラップします。ドキュメントは次のとおりです。 https://www.tensorflow.org/performance/xla/operation_semantics#sort 。
パッド<T はTTypeを拡張 >	XLA Pad オペレーターをラップします。ドキュメントは次のとおりです。 https://www.tensorflow.org/performance/xla/operation_semantics#pad 。
Recv <T はTTypeを拡張 >	別の XLA 計算から名前付きテンソルを受け取ります。
レプリカID	レプリカID。
SelfAdjointEig <T はTTypeを拡張 >	自己共役行列のバッチの固有分解を計算します (注: 実数入力のみがサポートされています)。
送信	名前付きテンソルを別の XLA 計算に送信します。
シャーディング<T extends TType >	指定されたシャーディング属性に基づいて入力をシャーディングする操作。
並べ替え<T extends TType >	XLA ソート演算子をラップします。ドキュメントは次のとおりです。 https://www.tensorflow.org/performance/xla/operation_semantics#sort 。
Svd <T はTTypeを拡張 >	自己共役行列のバッチの固有分解を計算します (注: 実数入力のみがサポートされています)。
XlaRecvFromHost <T はTTypeを拡張 >	ホストからテンソルを受け取る操作。
XlaSendToHost	テンソルをホストに送信する操作。
XlaSetBound	指定された入力値の境界を Xla コンパイラへのヒントとして設定します。 同じ値を返します。