Ouwen / tensorflow_interpolation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TensorFlow Regular Interp

Build regular_interp Op

Setup Docker Container

You are going to build the op inside a Docker container. Pull the provided Docker image from TensorFlow's Docker hub and start a container.

  docker pull tensorflow/tensorflow:custom-op
  docker run -it tensorflow/tensorflow:custom-op /bin/bash

Inside the Docker container, clone this repository. The code in this repository came from the Adding an op guide.

git clone https://github.com/tensorflow/custom-op.git
cd custom-op

Build PIP Package

You can build the pip package with either Bazel or make.

With bazel:

  ./configure.sh
  bazel build build_pip_pkg
  bazel-bin/build_pip_pkg artifacts

With Makefile:

  make pip_pkg

Install and Test PIP Package

Once the pip package has been built, you can install it with,

pip install artifacts/*.whl

Acknowledgements

Thanks to the Tensorflow team for creating the example-op guide. Thanks to @dfm for the interp kernel logic.

About

License:Apache License 2.0


Languages

Language:Python 49.8%Language:C++ 32.9%Language:Shell 12.2%Language:Makefile 3.5%Language:Smarty 1.6%