用張量流中的相對路徑加載用戶定義的操作數

Question

如何用張量流中的相對路徑加載用戶定義的操作數？它絕對路徑正常工作。但我想用相對路徑來做。

我用這個命令來構建OP： 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 

形象，但下面的代碼不會正常工作。

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/user_ops bazel build -s --copt=-D_GLIBCXX_USE_CXX11_ABI=0 :zero_out.so 或： cd tensorflow/core bazel build -s --copt=-D_GLIBCXX_USE_CXX11_ABI=0 user_ops:zero_out.so