Cross Compiling Tensorflow 2.0 (general)

yxy171230 · July 30, 2020, 2:46pm

I am trying to cross-compile the general TensorFlow-2 for my AGX board, but could not find a similar build_aarch64_lib.sh (according to TensorFlow 2.0 cross-compilation for TX2 - #4 by AastaLLL) file for the general version in the source code. Could anyone elaborate on how to cross-compile the general TensorFlow-2?

AastaLLL · July 31, 2020, 3:20am

Hi,

Please check the file here:

github.com

tensorflow/tensorflow/blob/r2.0/tensorflow/lite/tools/make/build_aarch64_lib.sh

#!/bin/bash
# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
#
# Licensed 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 KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================

set -x
set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

This file has been truncated. show original

Thanks.

yxy171230 · July 31, 2020, 2:25pm

Hi,
Thank you for the reply, This file cross-compiles the tensorflow lite, my question was to cross-compile the general version. The general version uses bazel instead of make for its compile. Is there a way to cross compile the general tensorflow?
Thanks!

AastaLLL · August 3, 2020, 8:15am

Hi,

You can check this with TensorFlow team to see if they support cross-compiling with bazel here:

Thanks.