問題タブ [tensorflow-xla]

XlaOpネイティブ C++ 値からを作成するための関数が多数あります。それぞれを使用してグラフを作成する方法を理解しようとしています。xla_builder.h を調べて、いくつかの候補を選び出し、オーバーロードと便利なラッパーを省略しました。有力候補は以下の2つだそうです

Parameterは「シンボル」用でConstantLiteralあり、定数値用であると考えるのは正しいですか? たとえば、 では、としてf(x) = x + 1エンコードします。1ConstantLiteralx

1. C++ 関数として記述f(x)し、アプリケーション サイトConstantLiteralで の値に別の関数を使用するx、または
2. xを使用してエンコードし、対応Parameterする から を構築します。そうは言っても、複数の s afaictで動作しない以外の を使用してを実際に呼び出す方法については明確ではありません。XlaComputationXlaBuilderXlaComputationLiteralLocalClientXlaComputation

これら2つのアプローチの違いは何ですか? 一方が他方よりも優れていますか？前者は、高階関数、つまりXlaComputations を受け入れる関数では使用できないように見えることに気付きました。

次は

• Infeedのストリーミング バージョンだと思いParameterます。
• Recvこれは、計算間でデータを渡す方法のように見えますが、実際にはまったく新しいものを作成するわけではありませんXlaOp。
• ReplicaId、Iota、およびXlaOp CreateToken(XlaBuilder* builder);この議論にはほとんど関係ないようです。

私はこれを正しく理解していますか？私が見逃した他の重要な機能はありますか?

XlaComputation別の を構築するときに を使用するにはどうすればよいXlaComputationですか? たとえば、次のようAddにしようとしている計算に組み込みたい

しかし、私は得る

エラーを処理する代わりに値をフェッチしようとしています無効な引数: ハンドル 3 の XlaOp はビルダー 'Add' によってビルドされていますが、ビルダー 'Mul' で使用しようとしています:

あるビルダーを別の計算で使用することになっていないことは承知していますが、それがその計算の結果に対応するXlaOp場合にそれを回避する方法がわかりません。XlaOpソースコードの次のコメントを読んだ後

XlaOp同じビルダーを使用する sにのみ適用されると思います。