diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2019-11-05 16:55:57 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-11-06 11:25:05 +0000 |
commit | 325279580fdadc36f82eb0aff2a2f3d72d5b71aa (patch) | |
tree | 81d690f9891d998ac22f71f61d704267b167bcbb /utils/GraphUtils.h | |
parent | 0fc2d7ad56b297852375b34322d0920193a21b8b (diff) | |
download | ComputeLibrary-325279580fdadc36f82eb0aff2a2f3d72d5b71aa.tar.gz |
COMPMID-2106: Extend get_input_accessor() to run input images through QASYMM8 networks
Change-Id: I5c44bcc95990c343d7829d885b93e8ff16691c50
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2226
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'utils/GraphUtils.h')
-rw-r--r-- | utils/GraphUtils.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index 826ffff8c3..bc0822cc89 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -486,7 +486,7 @@ inline std::unique_ptr<graph::ITensorAccessor> get_input_accessor(const arm_comp const std::string &image_file_lower = lower_string(image_file); if(arm_compute::utility::endswith(image_file_lower, ".npy")) { - return arm_compute::support::cpp14::make_unique<NumPyBinLoader>(image_file); + return arm_compute::support::cpp14::make_unique<NumPyBinLoader>(image_file, graph_parameters.data_layout); } else if(arm_compute::utility::endswith(image_file_lower, ".jpeg") || arm_compute::utility::endswith(image_file_lower, ".jpg") |