Загрузить определенный пользователем op с относительным путем в тензорном потоке

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