Sorry about that. The "adding an op" docs are updated at head (and will go live once 1.4 is released), but right now there's a bit of a mismatch. The important bits:

TF_LIB=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_lib())')

And add to the compiler flags:
-L$TF_LIB -ltensorflow_framework

It is also possible to build with --config=monolithic if you want to load custom ops with RTLD_GLOBAL in Python (should work just as it did pre-1.4).

@MarkDaoust maybe we should add a little warning on some of these docs that they're versioned, with a link to head?

Thanks, that fixed it! Looks like desired alignment in latest tf is 32
There used to be a time when TF docs had a pull-down that allowed to select version of docs, including "Master".

@allenlavoie

it did not work.

there is no libtensorflow_framework.so under $TL_LIB

My $TL_LIB is /home/tumh/.pyenv/versions/tensorflow/lib/python2.7/site-packages/tensorflow/core

I add both TF_INC and TF_LIB but still cannot fix it.
Does someone has same problem?

@twmht that library path comes from a version before TensorFlow 1.4; maybe you have a couple different Python installations with different TensorFlow versions?

@JuiHsuan-Kuo which TensorFlow version, and what's the error? Note that the updated documentation for 1.4+ is at https://www.tensorflow.org/versions/master/extend/adding_an_op

Are there relevant/accurate instructions for 1.3.0? I am seeing the same issue as @twmht with lack of tensorflow_framework from my 1.3.0 install.

If I do upgrade to 1.4.0 (which my codebase cannot support) I am able to build the shared object and avoid the missing kernel call but then I'm faced with

ImportError: libtensorflow_framework.so: cannot open shared object file: No such file or directory

@bearrito
https://www.tensorflow.org/versions/r1.3/extend/adding_an_op are the 1.3 instructions. https://www.tensorflow.org/versions/master/extend/adding_an_op are for master.

What's the context for that ImportError? Importing TensorFlow?

After messing around w/ the DYLD and LD lib paths, I ended up just brute forcing it....

bazel build --config=cuda //tensorflow/tools/lib_package:libtensorflow
cp bazel-bin/tensorflow/tools/lib_package/libtensorflow.tar.gz /opt/tensorlibs
tar xvfmz libtensorflow.tar.gz

And then just used linkopt

bazel build -c opt --config=cuda //dragnn/core:gen_dragnn_bulk_ops_py_wrappers_cc --verbose_failures --linkopt=/opt/tensorlibs/lib/libtensorflow_framework.so

How about the build instructions for GPU ops. Tensorflow tutorials even never mention it. @allenlavoie

I find one here, but it also meets the same problem

I'm using tf.14. I get the same error when running a custom op: undefined symbol: _ZTIN10tensorflow8OpKernelE

when I compile using this flag: -L$TF_LIB -ltensorflow_framework,

I get this error:
/usr/bin/ld: cannot find -ltensorflow_framework.
can someone help please?

I'm using tf.14. I get the same error when running a custom op: undefined symbol: _ZTIN10tensorflow8OpKernelE

when I compile using this flag: -L$TF_LIB -ltensorflow_framework,

I get this error:
/usr/bin/ld: cannot find -ltensorflow_framework.
can someone help please?

I am having the exact same issue

I resolved the issue after spending days of scraping the internet to compile a custom tf op. I am including the steps that I had to take for other people to use.

• downgrade your python version to 3.6
• downgrade your tf to 1.10.0
• use CUDA 9

@kasrayazdani, sorry I missed your first post. That does not sound like fun.

There appears to be a new repository for this:

https://github.com/tensorflow/custom-op

I will see if I can delete the old doc and just leave a pointer to this new repo.

yaroslavvb
please let us know if the comments above help resolve the issue.

TF : 1.13.2 (build from source) / cuda 10.0 , cudnn7.4.24.2 , gpu-driver-460
OS : ubuntu 18.4
gcc : 7.5.0
cmake : 3.22.1
python 3.6.9

My tensorflow construction order :

bazel build --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" //tensorflow:libtensorflow_cc.so //tensorflow/tools/pip_package:build_pip_package

I also made a custom op for myself, but
while trying to load the .so file I run into

_fast_oll = tf.load_op_library('obj_loader/cmake-build/libobj_loader.so')

File "/home/shyan/testtensorflow/azmoon/lib/python3.6/site-packages/tensorflow/python/framework/load_library.py", line 61, in load_op_library
lib_handle = py_tf.TF_LoadLibrary(library_filename)
tensorflow.python.framework.errors_impl.NotFoundError: obj_loader/cmake-build/libobj_loader.so: undefined symbol: _ZTIN10tensorflow8OpKernelE

Can anyone help me?

Hi @yaroslavvb ! Have you checked latest document on a registering a custom op from 2.8 version yet? Thanks!

This issue has been automatically marked as stale because it has no recent activity. It will be closed if no further activity occurs. Thank you.

Closing as stale. Please reopen if you'd like to work on this further.

@anirudh161 has been working on this with some of our engineers.

Ok @MarkDaoust ! Reopening this issue then. Thank you!

Marking this issue as resolved as there has been no activity on it. Please open a new issue if you're facing issues with creating custom ops. Thanks!

Are you satisfied with the resolution of your issue?
Yes
No