custom-op
custom-op copied to clipboard
tensorflow
Guide for building custom op for TensorFlow
Hi there! Is there any example where Custom OP written using C API was compiled successfully and executed after tf_load_library()? I have referred the entire kernels.h ,ops.h and c_api.h files...
Need a working example how to build the custom ops for new tensorflow versions - like 2.8. Current docker container come with old python version (3.6) and won't even allow...
Hi, I built the CPU custom ops sample following the instructions for the Docker based build process but when I tried to run the op, it fails in the loading...
To reproduce: 1. git clone, cd and Run docker: `docker run -it --rm --gpus all -w /workspace/ -v ${PWD}:/workspace/ tensorflow/tensorflow:custom-op-gpu-ubuntu16` 2. Attempt to build the op: `make time_two_op` Will result...
My project has to depend on both shared library libtensorflow_cc.so and libtensorflow_framework.so So I tried to edit the tf/BUILD.tpl file and added another rule as following `` cc_library( name =...
The guide says to use nvidia-docker but this isn't possible for windows distributions. Is there an alternative?
I use the Makefile to generate the .so file. I tried to load the custom op library, but I get the following error: 
I used a custom op for data transform and got a saved model. All above is ok Now I want to load this model and do some predict. Code likes...
The link from the text below is broken. **Extend and Test the Op in Python** Once you have built your custom op shared library, you can follow the example in...
