tensorflowでconv_opsのような演算を作成するには？

Question

私は

をやろうとしている私はC++とbazelに新たなんだと私はtensorflowで畳み込み演算にいくつかの変更をしたいので、私は私の最初のステップは、ちょうどOPSを作成することであると判断いいね。私は

を行っている何

私は// tensorflow /コア/カーネルからconv_ops.ccをコピーして、私のnew_conv_ops.ccにregistrated OPSの名前を変更します。重複を避けるために、ファイル内の関数の名前を変更しました。ここに私のBUILDがありますfile.

ご覧のとおり、私はconv_opsのdeps属性を// tensorflow/core/kernels/BUILDからコピーします。次に、私は "bazel build -c opt //tensorflow/core/user_ops:new_conv_ops.so"を使って新しいopをビルドします。

は私の問題は

ある何それから私は私がbounds_checkを削除しようとしましたし、次のdepsのために同じエラーを得た。このerror.

を得ました。それから、// tensorflow/core/user_opsの//テンソルフロー/コア/カーネルにhファイルを含めるにはいくつかの問題があることに気付きました。だから私はconv_opsのようなexcatcly新しいopを完璧に作成できますか？

Answer 1

TensorFlowにカスタム操作を追加する方法については、チュートリアルhereを参照してください。実際のコードexamplesを見ることもできます。

特定の問題に対処するには、tf_custom_op_libraryマクロがターゲットに必要な依存性のほとんどを追加することに注意してください。次のように書くことができます：

tf_custom_op_library(
    name="new_conv_ops.so", 
    srcs=["new_conv_ops.cc"] 
)