1 | /* |
2 | * SPDX-License-Identifier: Apache-2.0 |
3 | */ |
4 | |
5 | // Helper Methods for Adapters |
6 | |
7 | #pragma once |
8 | |
9 | #include "onnx/common/ir.h" |
10 | |
11 | namespace ONNX_NAMESPACE { |
12 | namespace version_conversion { |
13 | int check_numpy_unibroadcastable_and_require_broadcast( |
14 | const std::vector<Dimension>& input1_sizes, |
15 | const std::vector<Dimension>& input2_sizes); |
16 | |
17 | void assert_numpy_multibroadcastable( |
18 | const std::vector<Dimension>& input1_sizes, |
19 | const std::vector<Dimension>& input2_sizes); |
20 | |
21 | void assertNotParams(const std::vector<Dimension>& sizes); |
22 | |
23 | void assertInputsAvailable(const ArrayRef<Value*>& inputs, const char* name, uint64_t num_inputs); |
24 | } // namespace version_conversion |
25 | } // namespace ONNX_NAMESPACE |
26 | |