相対パスをテンソルでロードしてユーザ定義演算をロード

Question

ユーザ定義演算を相対パスでテンソルフローにロードするにはどうすればよいですか？絶対パスで正常に動作します。しかし、私はそれを相対パスを使ってやりたい。

私はオペアンプを構築するために、このコマンドを使用： bazel build -s --copt="-D_GLIBCXX_USE_CXX11_ABI=0" -c opt //tensorflow/core/user_ops:zero_out.so

私が使用したビルドファイルは次のとおりです。

load("//tensorflow:tensorflow.bzl", "tf_custom_op_library") tf_custom_op_library( name = "zero_out.so", srcs = ["zero_out.cc"], )

以下のコードは動作しません。イメージが見つかりませんでしたエラーを示します。 tensorflow.python.framework.errors.NotFoundError：のdlopen（zero_out.so、6）：画像

import os.path 
import tensorflow as tf 
_zero_out_module = tf.load_op_library(os.path.join(tf.resource_loader.get_data_files_path(),'zero_out.so')) 
zero_out = _zero_out_module.zero_out 

を見つけていないが、以下のコードが正常に動作しません。あなたは、例えば、bazelに cd tensorflow/core/user_ops bazel build -s --copt=-D_GLIBCXX_USE_CXX11_ABI=0 :zero_out.so または相対パスを使用することができ

import os.path 
import tensorflow as tf 
_zero_out_module = tf.load_op_library('/Users/sahilsingla/tensorflow/bazel-bin/tensorflow/core/user_ops/zero_out.so') 
zero_out = _zero_out_module.zero_out 

Answer 1

： cd tensorflow/core bazel build -s --copt=-D_GLIBCXX_USE_CXX11_ABI=0 user_ops:zero_out.so