1 | /* |
---|---|
2 | * SPDX-License-Identifier: Apache-2.0 |
3 | */ |
4 | |
5 | #pragma once |
6 | |
7 | #include "onnx/onnx-operators_pb.h" |
8 | |
9 | namespace ONNX_NAMESPACE { |
10 | |
11 | template <typename T> |
12 | TensorProto ToTensor(const T& value); |
13 | |
14 | template <typename T> |
15 | TensorProto ToTensor(const std::vector<T>& values); |
16 | |
17 | template <typename T> |
18 | const std::vector<T> ParseData(const TensorProto* tensor_proto); |
19 | |
20 | } // namespace ONNX_NAMESPACE |
21 |