Refactor tensorflow in Dockerfile

Author: madelen-at-work

Dockerfile

@@ -9,6 +9,17 @@ FROM arm64v8/ubuntu:${UBUNTU_VERSION} AS containerized_aarch64
 FROM arm32v7/ubuntu:${UBUNTU_VERSION} AS containerized_armv7hf
 
 FROM ${REPO}/acap-native-sdk:${VERSION}-${ARCH}-ubuntu${UBUNTU_VERSION} AS acap-native-sdk
+
+FROM acap-native-sdk AS tensorflow
+
+WORKDIR /opt
+
+# Get TensorFlow and TensorFlow Serving - doing this in parallell saves a little bit of time
+RUN <<EOF
+    git clone -b r2.9 https://github.com/tensorflow/tensorflow.git /opt/tensorflow/tensorflow
+    git clone -b r2.9 https://github.com/tensorflow/serving.git /opt/tensorflow/serving
+EOF
+
 FROM acap-native-sdk AS build
 
 ARG ARCH
@@ -151,11 +162,8 @@ RUN <<EOF
         "$SDKTARGETSYSROOT"/usr/include
 EOF
 
-# Get TensorFlow and TensorFlow Serving
-RUN <<EOF
-    git clone -b r2.9 https://github.com/tensorflow/tensorflow.git /opt/tensorflow/tensorflow
-    git clone -b r2.9 https://github.com/tensorflow/serving.git /opt/tensorflow/serving
-EOF
+# Get TensorFlow and TensorFlow Serving that we've pulled in parallell
+COPY --from=tensorflow /opt/tensorflow /opt/tensorflow
 
 ## Setup build structure
 WORKDIR /opt/app