如何用張量流中的相對路徑加載用戶定義的操作數?它絕對路徑正常工作。但我想用相對路徑來做。用張量流中的相對路徑加載用戶定義的操作數
我用這個命令來構建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
我編輯的問題。我不認爲這很清楚。 –