RSTensorFlow is a modified version of TensorFlow that utilizes the GPUs of commodity Android devices. RSTensorFlow is developed by the Networked and Embedded Systems Lab (NESL) at UCLA.
git clone https://github.com/nesl/RSTensorFlow.git
You need to update the WORKSPACE file to update the Android SDK/NDK location path.
master branch is the original TensorFlow libraries. matmul_only branch has code for GPU-accelerated matrix multiplication only. matmul_and_conv branch has the code for GPU-accelerated matrix multiplication and convolution operations.For example,
git checkout matmul_only
Build the Android demo app
or build the TensorFlow for Android libraries.
bazel build -c opt //tensorflow/examples/android:tensorflow_demo
Then, build the Java library
bazel build -c opt //tensorflow/contrib/android:libtensorflow_inference.so \ --crosstool_top=//external:android/crosstool \ --host_crosstool_top=@bazel_tools//tools/cpp:toolchain \ --cpu=armeabi-v7a
bazel build //tensorflow/contrib/android:android_tensorflow_inference_java
Project developed by NESL, currently maintained by Moustafa Alzantot