sw/vec_iree: Add support for importing .raw input file Change-Id: Ia8ccf899e1abd41b0fec56e774899831f00e4bb6
diff --git a/build_tools/gen_mlmodel_input.py b/build_tools/gen_mlmodel_input.py index bbcd8d0..e40c8f4 100755 --- a/build_tools/gen_mlmodel_input.py +++ b/build_tools/gen_mlmodel_input.py
@@ -59,7 +59,7 @@ raise ValueError("Input shape < 3 dimensions") input_ext = os.path.splitext(input_name)[1] is_audio = False - if (not input_ext) or (input_ext == '.bin'): + if (not input_ext) or (input_ext == '.bin') or (input_ext == '.raw'): with open(input_name, mode='rb') as f: input = np.fromfile(f, dtype=np.uint8 if is_quant else np.float32) input = input[:np.prod(input_shape)].reshape(np.prod(input_shape))