Skip to main content

Missing layers, ops & etc. for TensorFlow

Project description

tfmiss

The missing OPs, layer & etc. for TensorFlow

Development

Environment

Install all dependencies including python headers. Do not use pyenv on MacOS X, otherwise tests mostly likely will fail.

Build PIP package manually

You can build the pip package with Bazel v0.25.3:

# GPU support
export TF_NEED_CUDA="1"

# Set these if the below defaults are different on your system
export TF_CUDA_VERSION="11.2"
export TF_CUDNN_VERSION="8"
export CUDA_TOOLKIT_PATH="/usr/local/cuda"
export CUDNN_INSTALL_PATH="/usr/lib/x86_64-linux-gnu"

./configure.py
bazel clean --expunge
bazel test --test_output=errors //tfmiss/...
bazel build build_pip_pkg
bazel-bin/build_pip_pkg wheels

Build release with Linux docker container

# Requires about 4Gb of RAM allocated to Docker
DOCKER_BUILDKIT=1 docker build -t miss --output type=local,dest=wheels --build-arg PY_VERSION=3.8 ./

Install and test PIP package

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

pip install wheels/*.whl

Now you can test out the pip package:

cd /
python -c "import tensorflow as tf;import tfmiss as tfm;print(tfm.text.zero_digits('123').numpy())"

You should see the op zeroed out all nonzero digits in string "123":

000

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release. See tutorial on generating distribution archives.

Built Distributions

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page