From e770f7b0cff0684cee9829da18f133cbb6a2262e Mon Sep 17 00:00:00 2001 From: Alexander Peskov Date: Thu, 22 Oct 2020 02:30:42 +0300 Subject: [PATCH] [TEST] improve data utils Signed-off-by: Alexander Peskov --- .../tests/ie_test_utils/common_test_utils/data_utils.cpp | 10 ++++++++++ .../tests/ie_test_utils/common_test_utils/data_utils.hpp | 8 ++++++++ 2 files changed, 18 insertions(+) diff --git a/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.cpp b/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.cpp index 12ae4ee..4867d8d 100644 --- a/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.cpp +++ b/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.cpp @@ -104,6 +104,10 @@ void fill_data_with_broadcast(InferenceEngine::Blob::Ptr& blob, InferenceEngine: auto src_ptr = get_data(values); switch (blob->getTensorDesc().getPrecision()) { + case InferenceEngine::Precision::U64: + case InferenceEngine::Precision::I64: + copy_7D(src_ptr, src_strides, dst_ptr, dst_strides, dst_dims); + break; case InferenceEngine::Precision::FP32: case InferenceEngine::Precision::I32: copy_7D(src_ptr, src_strides, dst_ptr, dst_strides, dst_dims); @@ -189,6 +193,12 @@ InferenceEngine::Blob::Ptr make_reshape_view(const InferenceEngine::Blob::Ptr &b return new_blob; } +size_t byte_size(const InferenceEngine::TensorDesc &tdesc) { + auto prc = tdesc.getPrecision(); + auto dims = tdesc.getDims(); + return prc.size() * std::accumulate(std::begin(dims), std::end(dims), (size_t)1, std::multiplies()); +} + /** * repeated filling tensor with data. * diff --git a/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.hpp b/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.hpp index c6eeab1..8d46d85 100644 --- a/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.hpp +++ b/inference-engine/tests/ie_test_utils/common_test_utils/data_utils.hpp @@ -72,6 +72,14 @@ InferenceEngine::Blob::Ptr make_reshape_view(const InferenceEngine::Blob::Ptr &b */ void fill_data_const(InferenceEngine::Blob::Ptr& blob, float val); + +/** + * Calculate size of buffer required for provided tensor descriptor. + * @param tdesc provided tensor descriptor + * @return size in bytes + */ +size_t byte_size(const InferenceEngine::TensorDesc &tdesc); + static void fill_data_bbox(float *data, size_t size, int height, int width, float omega) { float center_h = (height - 1.0f) / 2; float center_w = (width - 1.0f) / 2; -- 2.7.4