Skip to content

Commit 7f3d0a9

Browse files
committed
Merge remote-tracking branch 'upstream/main' into merge_main_to_dev_zephyr
2 parents 685d5a1 + 8414a20 commit 7f3d0a9

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

core/iwasm/libraries/wasi-nn/src/wasi_nn_tensorflowlite.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,11 @@ set_input(void *tflite_ctx, graph_execution_context ctx, uint32_t index,
281281
{
282282
TFLiteContext *tfl_ctx = (TFLiteContext *)tflite_ctx;
283283

284+
if (input_tensor->type != fp32) {
285+
NN_ERR_PRINTF("unsupported input tensor type %u", input_tensor->type);
286+
return runtime_error;
287+
}
288+
284289
wasi_nn_error res;
285290
if (success != (res = is_valid_graph_execution_context(tfl_ctx, ctx)))
286291
return res;

core/iwasm/libraries/wasi-nn/test/build.sh

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,17 @@
33
# Copyright (C) 2019 Intel Corporation. All rights reserved.
44
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
55

6+
# on intel mac, this ends up with a lot of the following error.
7+
#
8+
# AttributeError: 'Sequential' object has no attribute '_get_save_spec'.
9+
#
10+
# * "pip install tensorflow" installs tensorflow 2.16.2 on intel mac.
11+
# (because it's the last version before tf deprecated the target.)
12+
# * keras 3 support in the version seems incomplete (thus the error)
13+
# * a workaround: use keras 2 as mentioned in:
14+
# https://github.com/tensorflow/tensorflow/releases/tag/v2.16.1
15+
# https://blog.tensorflow.org/2024/03/whats-new-in-tensorflow-216.html
16+
617
CURR_PATH=$(cd $(dirname $0) && pwd -P)
718

819
# WASM application that uses WASI-NN

0 commit comments

Comments
 (0)