Spaces:
Sleeping
Sleeping
File size: 2,126 Bytes
dc2106c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
// Copyright (c) ONNX Project Contributors
/*
* SPDX-License-Identifier: Apache-2.0
*/
#include <memory>
#include <string>
#include <utility>
#include <vector>
#include "onnx/onnx_pb.h"
namespace ONNX_NAMESPACE {
namespace inliner {
// IR version 10 introduces overloaded function names. The following APIs to specify
// functions to be inlined currently allow only specifying (domain, name). Thus,
// either all overloads of a function are inlined or none.
// The older-style ids are used below for backward compatibility.
// A FunctionId is a pair of strings (domain, function name).
using FunctionId = std::pair<std::string, std::string>;
// A vector of FunctionIds.
using FunctionIdVector = std::vector<FunctionId>;
// Interface used to represent a set of function ids for the inliner.
class FunctionIdSet {
public:
virtual bool Contains(const std::string& function_domain, const std::string& function_name) const = 0;
virtual ~FunctionIdSet() = default;
// Factory methods for creating FunctionIdSet instances.
// Creates a set representing the elements in the given vector, if invert is false.
// Otherwise, creates a set representing elements not in the given vector.
static std::unique_ptr<FunctionIdSet> Create(FunctionIdVector&& function_ids, bool invert = false);
};
// Inlines the model-local functions in the given model that are in the given set.
// The inlined functions are removed from the model's list of functions as well.
void InlineSelectedFunctions(ModelProto& model, const FunctionIdSet& to_inline);
// Inlines all model-local functions in the given model. This supports version
// conversion, an advanced feature that is not enabled by default. When enabled,
// the inliner will attempt to convert the version of the inlined function to
// match the version of the model. If not enabled, the inliner will only inline
// functions that use opset versions that are compatible with the model.
void InlineLocalFunctions(ModelProto& model, bool convert_version = false);
} // namespace inliner
} // namespace ONNX_NAMESPACE
|