Невозможно создать окончательный двоичный файл с использованием AOT-компиляции tenorflow для серверной части процессора

Я следовал за учебником здесь: TensorFlow AOT сборник

В соответствии с шагами 1 и 2 я скомпилировал подграф и сгенерировал заголовок (test_graph_tfmatmul.h) и объект (test_graph_tfmatmul.o) файлы;

Согласно шагу 3, я использовал пример кода (названный как my_code.cc) вызвать подграф;

Согласно шагу 4, я добавил фрагмент кода cc_binary к существующему BUILD файл (//tensorflow/compiler/aot/tests/BUILD) и попытался создать окончательный двоичный файл с помощью команды:

bazel build //tensorflow/compiler/aot/tests:my_binary

но я получил следующую ошибку:

undeclared inclusion(s) in rule '//tensorflow/compiler/aot/tests:my_binary':
this rule is missing dependency declarations for the following files included by 'tensorflow/compiler/aot/tests/tfcompile_test.cc':
'/home/tensorFlow_src/tensorflow/tensorflow/compiler/aot/tests/test_graph_tfmatmul.h'

Любой совет приветствуется. Благодарю.

1

Решение

Эта проблема, наконец, выясняется с помощью tf_library построить cc_library на шаге 2 вместо использования tfcompile непосредственно. т.е., tf_library побежит tfcompile генерировать заголовок и объектные файлы.

Пожалуйста, обратитесь к https://github.com/tensorflow/tensorflow/issues/13482 для более подробной информации.

1

Другие решения

Базель жалуется, что в tfcompile_test.cc вы #include "tensorflow/compiler/aot/tests/test_graph_tfmatmul.h", но нет никакой зависимости, обеспечивающей этот заголовок, объявленный в файле BUILD. Вы добавили ":test_graph_tfmatmul" в глубину my_binary?

0

По вопросам рекламы [email protected]