diff options
Diffstat (limited to 'source/use_case/inference_runner/src/UseCaseHandler.cc')
-rw-r--r-- | source/use_case/inference_runner/src/UseCaseHandler.cc | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/source/use_case/inference_runner/src/UseCaseHandler.cc b/source/use_case/inference_runner/src/UseCaseHandler.cc index a75b2e4..b98b1c5 100644 --- a/source/use_case/inference_runner/src/UseCaseHandler.cc +++ b/source/use_case/inference_runner/src/UseCaseHandler.cc @@ -41,6 +41,17 @@ namespace app { const size_t numInputs = model.GetNumInputs(); +#if VERIFY_TEST_OUTPUT + info("Initial input tensors values:\n"); + for (size_t inputIndex = 0; inputIndex < model.GetNumInputs(); inputIndex++) { + arm::app::DumpTensor(model.GetInputTensor(inputIndex)); + } + info("Initial output tensors values:\n"); + for (size_t outputIndex = 0; outputIndex < model.GetNumOutputs(); outputIndex++) { + arm::app::DumpTensor(model.GetOutputTensor(outputIndex)); + } +#endif /* VERIFY_TEST_OUTPUT */ + /* Populate each input tensor with random data. */ for (size_t inputIndex = 0; inputIndex < numInputs; inputIndex++) { |