onnx/version_converter/helper.h

0001 // Copyright (c) ONNX Project Contributors
0002 //
0003 // SPDX-License-Identifier: Apache-2.0
0004
0005 // Helper Methods for Adapters
0006
0007 #pragma once
0008
0009 #include <vector>
0010
0011 #include "onnx/common/ir.h"
0012
0013 namespace ONNX_NAMESPACE {
0014 namespace version_conversion {
0015 int check_numpy_unibroadcastable_and_require_broadcast(
0016     const std::vector<Dimension>& input1_sizes,
0017     const std::vector<Dimension>& input2_sizes);
0018
0019 void assert_numpy_multibroadcastable(
0020     const std::vector<Dimension>& input1_sizes,
0021     const std::vector<Dimension>& input2_sizes);
0022
0023 void assertNotParams(const std::vector<Dimension>& sizes);
0024
0025 void assertInputsAvailable(const ArrayRef<Value*>& inputs, const char* name, uint64_t num_inputs);
0026 } // namespace version_conversion
0027 } // namespace ONNX_NAMESPACE