Support ARM64 platform in TensorFlow examples (#2119)

* Support ARM64 platform in TensorFlow examples

Signed-off-by: akhilsaivenkata <[email protected]>

* Support ARM64 platform in TensorFlow examples

Signed-off-by: akhilsaivenkata <[email protected]>

* Support ARM64 platform in TensorFlow examples

Signed-off-by: akhilsaivenkata <[email protected]>

---------

Signed-off-by: akhilsaivenkata <[email protected]>

Author: akhilsaivenkata

.github/workflows/publish-example-images.yaml

@@ -22,14 +22,19 @@ jobs:
       matrix:
         include:
           - component-name: tf-dist-mnist-test
+            platforms: linux/amd64,linux/arm64
             dockerfile: examples/tensorflow/dist-mnist/Dockerfile
           - component-name: tf-distributed-worker
+            platforms: linux/amd64,linux/arm64
             dockerfile: examples/tensorflow/distribution_strategy/estimator-API/Dockerfile
           - component-name: tf-multi-worker-strategy
+            platforms: linux/amd64,linux/arm64
             dockerfile: examples/tensorflow/distribution_strategy/keras-API/Dockerfile
           - component-name: tf-mnist-with-summaries
+            platforms: linux/amd64,linux/arm64
             dockerfile: examples/tensorflow/mnist_with_summaries/Dockerfile
           - component-name: tf-smoke
+            platforms: linux/amd64,linux/arm64
             dockerfile: examples/tensorflow/tf_sample/Dockerfile
           - component-name: pytorch-dist-sendrecv-test
             platforms: linux/amd64,linux/arm64

examples/tensorflow/distribution_strategy/keras-API/Dockerfile

@@ -1,5 +1,10 @@
 FROM python:3.9
 
+RUN apt-get update && \
+    apt-get install -y libhdf5-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
 RUN pip install tensorflow==2.11.0 tensorflow_datasets==4.7.0
 
 COPY examples/tensorflow/distribution_strategy/keras-API/multi_worker_strategy-with-keras.py /