Добавить новый оператор в tenorflow на Google Colab

Question

Я пытаюсь скомпилировать некоторые файлы.cc в Google Colab (как показано здесь и здесь), но я не могу найти файл op.h, чтобы включить, я получаю следующую ошибку при компиляции:

word2vec_ops.cc:16:10: fatal error: tensorflow/core/framework/op.h: No such file or directory
#include "tensorflow/core/framework/op.h"
      ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
compilation terminated.
word2vec_kernels.cc:16:10: fatal error: tensorflow/core/framework/op.h: No such file or directory
#include "tensorflow/core/framework/op.h"
      ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
compilation terminated.

Answer 1

Вы можете скомпилировать файлы.cc в Google Colab с помощью следующей команды gcc:

$ gcc -Wl,-R/path/to/lib -I/path/to/include -L/path/to/lib -o myAppName mycode.c -llibapp2

найдите свою библиотеку, включив путь

import tensorflow as tf
tf.sysconfig.get_include()

Пример:

!g++ -std=c++11 -I /usr/local/lib/python3.6/dist-packages/tensorflow_core/include -shared word2vec_ops.cc word2vec_kernels.cc -o word2vec_ops.so -fPIC ${TF_CFLAGS[@]} ${TF_LFLAGS[@]} -O2```