custom-op
custom-op copied to clipboard
Guide for building custom op for TensorFlow
Hi, I have problems on building the official example with GPU support with bazel in tensorflow source folder. I use tf_custom_op_library rule defined in //tensorflow:tensorflow.bzl using nvcc + g++ works...
Since https://github.com/tensorflow/tensorflow/commit/7efc61175c540a56b03e829ec917ce9efc1f06f9 custom-op's configure script will no longer properly parse the shared library name for macos TensorFlow: https://github.com/tensorflow/custom-op/blob/master/configure.sh#L94 As a quick fix in Addons we just [altered the string parsing...
Fixed typos and broken links in documentation.
custom-op project itself is out of sync with the original TF repository, so trying to upgrade it to the recent one (v2.11.0)