nn: fix compatability with tensorflow 2.4.0
BUG=chromium:1162396
TEST=FEATURES=test emerge-betty aosp-frameworks-ml-nn
Cq-Depend: chromium:2607025
Change-Id: I643dd554a24f1a50923d4edcb5211a8d1d5e45f4
Reviewed-on: https://chromium-review.googlesource.com/c/aosp/platform/frameworks/ml/+/2612994
Tested-by: Jim Pollock <jmpollock@chromium.org>
Tested-by: Xinglong Luan <alanlxl@chromium.org>
Auto-Submit: Jim Pollock <jmpollock@chromium.org>
Reviewed-by: Xinglong Luan <alanlxl@chromium.org>
Reviewed-by: Michael Pishchagin <mblsha@google.com>
Commit-Queue: Xinglong Luan <alanlxl@chromium.org>
2 files changed