custom-op
custom-op copied to clipboard
tensorflow
Guide for building custom op for TensorFlow
I don't know if this issue is related to version of protocol buffer used in the current existing image `tensorflow/tensorflow:2.2.0-custom-op-gpu-ubuntu16` to build custom ops. I'm trying to build GRPC ops...
The following error is reported when compiling: `$make time_two_op` `nvcc -std=c++11 -c -o tensorflow_time_two/python/ops/_time_two_ops.cu.o tensorflow_time_two/cc/kernels/time_two_kernels.cu.cc -I/usr/local/lib/python3.6/dist-packages/tensorflow_core/include -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -O2 -std=c++11 -L/usr/local/lib/python3.6/dist-packages/tensorflow_core -l:libtensorflow_framework.so.1 -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -DNDEBUG --expt-relaxed-constexpr...
I try to build the example ops in this repo as a pip package on Windows 10. I followed the instructions [here](https://www.tensorflow.org/install/source_windows) to set up the requirements using the given...
**System information** - OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Windows 10 - TensorFlow version: 2.1 - Python version: 3.6 - Bazel version (if compiling from source): 3.7.0 -...
I have the PID while run python : 10495. What i want to do: ``` gdb -p 10495 (gdb)break tensorflow_zero_out/cc/kernels/zero_out_kernels.cc:33 ``` And my question is how to compile custom op...
Hi, i already used your zero out custom op successfully , now i am trying to build a fastertransformer custom op accroding to this.https://github.com/NVIDIA/DeepLearningExamples/tree/master/FasterTransformer/v1/fastertransformer/tf_op. However, i feel it's pretty hard...
Hi, i am building zero_out customer op following ur readme file. However, when i was using  this command to build, i got this error below:  My bazel version...
I know that in your project description, you mentioned that one has to download the docker image provided in order to build the custom op successfully. I first tried to...
