Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[TFLite Runtime] Add TFLite Runtime dependencies to CI CPU docker build #5437

Merged
merged 10 commits into from
Apr 27, 2020
Merged

[TFLite Runtime] Add TFLite Runtime dependencies to CI CPU docker build #5437

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
9db661b
Build ci cpu image with TFLITE ON
michalpiszczek Apr 23, 2020
5c4d089
Enable test for tflite remote execution via rpc
michalpiszczek Apr 23, 2020
77cac8d
Add tensorflow dep to ci cpu image
michalpiszczek Apr 24, 2020
47b47b1
Make tflite unit tests findable by pytest
michalpiszczek Apr 24, 2020
cc4d55b
Add newlines where appropriate
michalpiszczek Apr 24, 2020
921ca28
Re-word pytest skip annotation
michalpiszczek Apr 24, 2020
9bd7261
Revert test file
michalpiszczek Apr 24, 2020
15230ce
Fix whitespace
michalpiszczek Apr 24, 2020
27eeed5
Set TFLite off in cpu config
michalpiszczek Apr 24, 2020
abd7ffc
Build tflite runtime with latest stable tf
michalpiszczek Apr 24, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions docker/Dockerfile.ci_cpu
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,3 +63,11 @@ RUN bash /install/ubuntu_install_antlr.sh
# Chisel deps for TSIM
COPY install/ubuntu_install_chisel.sh /install/ubuntu_install_chisel.sh
RUN bash /install/ubuntu_install_chisel.sh

# TFLite deps
COPY install/ubuntu_install_tflite.sh /install/ubuntu_install_tflite.sh
RUN bash /install/ubuntu_install_tflite.sh

# TensorFlow deps
COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh
RUN bash /install/ubuntu_install_tensorflow.sh
12 changes: 9 additions & 3 deletions docker/install/ubuntu_install_tflite.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
#
# http://www.apache.org/licenses/LICENSE-2.0
#
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
Expand All @@ -26,12 +26,18 @@ cd flatbuffers
cmake -G "Unix Makefiles" -DCMAKE_BUILD_TYPE=Release
make install -j8
cd ..
rm -rf flatbuffers

# Install flatbuffers python packages.
pip3 install flatbuffers
pip2 install flatbuffers

# Build the TFLite static library, necessary for building with TFLite ON.
# The library is built at:
# tensorflow/tensorflow/lite/tools/make/gen/*/lib/libtensorflow-lite.a.
git clone https://github.com/tensorflow/tensorflow --branch=r2.1
./tensorflow/tensorflow/lite/tools/make/download_dependencies.sh
./tensorflow/tensorflow/lite/tools/make/build_lib.sh

# Setup tflite from schema
mkdir tflite
cd tflite
Expand Down