blob: 4ed6543d287816bec5bf134a38f488b03d5a6c1c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
//
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
namespace armnnUtils
{
std::string ConstructTensorShapeString(const std::vector<int>& shape)
{
std::string shapeStr;
for (int i : shape)
{
shapeStr = fmt::format("{} dim {{ dim_value: {} }}", shapeStr, i);
}
return shapeStr;
}
} // namespace armnnUtils
|