mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-17 01:44:45 +00:00
4b875e3543
* initial fix * refactor the function handle * update the implementation * fix linux build break * fix training build * fix minmal build * fix gradient checker * deprecate the local function members in graph. host it in model * fix changming's comments * fix comments about inlined containers * fix a missed inlined container * fix training build * avoid const for std string_view Co-authored-by: Cheng Tang <chenta@microsoft.com>
57 lines
2.8 KiB
C++
57 lines
2.8 KiB
C++
#pragma once
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "core/common/common.h"
|
|
#include "onnx/onnx_pb.h"
|
|
#include "core/graph/graph.h"
|
|
#include "core/graph/indexed_sub_graph.h"
|
|
#include "core/graph/function.h"
|
|
#include "core/graph/schema_registry.h"
|
|
|
|
namespace onnxruntime {
|
|
namespace function_utils {
|
|
|
|
/** Create a OpSchema given a subgraph EP whant to fuse.
|
|
* This is used when EP return fusion in GetCapability implementation.
|
|
* @param graph The graph which host the subgraph.
|
|
* @param nodes_to_fuse The metadata for the subgraph that EP want to fuse.
|
|
*/
|
|
std::unique_ptr<ONNX_NAMESPACE::OpSchema> CreateSchema(const Graph& graph,
|
|
const IndexedSubGraph& nodes_to_fuse);
|
|
|
|
/** Create a OpSchema given from a local function in onnx model.
|
|
* @param function_domain The domain of the function.
|
|
* @param function_name The name of the function.
|
|
* @param model_local_functions The map of local functions in the same onnx model.
|
|
* This will be used as context for the function's type/shape inference.
|
|
* @param domain_version_map Domain to version map used in current onnx model.
|
|
* @param schema_registry The schema registry current model is using.
|
|
* @param logger The logger current model is using.
|
|
* @param allow_released_opsets_only The flag whether we only enable released opset.
|
|
*/
|
|
std::unique_ptr<ONNX_NAMESPACE::OpSchema> CreateSchema(const std::string& function_domain,
|
|
const std::string& function_name,
|
|
const InlinedHashMap<std::string, const ONNX_NAMESPACE::FunctionProto*>& model_local_functions,
|
|
const std::unordered_map<std::string, int>& domain_version_map,
|
|
const SchemaRegistryManager& schema_registry,
|
|
const logging::Logger& logger,
|
|
bool allow_released_opsets_only);
|
|
|
|
/** Get the unique id for function. This is used as a key to find the
|
|
* relevant model local function from it's container.
|
|
* @param function_domain Domain for the function.
|
|
* @param function_name Name of the function. Name should match the OpType of the node which references the function.
|
|
*/
|
|
inline std::string GetFunctionIdentifier(std::string_view function_domain, std::string_view function_name) {
|
|
return function_domain.data() + std::string(":") + function_name.data();
|
|
}
|
|
|
|
Status Instantiate(onnxruntime::Graph& graph,
|
|
const onnxruntime::NodeIndex node_index,
|
|
const ONNX_NAMESPACE::FunctionProto& onnx_func_proto,
|
|
std::unique_ptr<Function>& output);
|
|
|
|
}
|
|
|
|
} // namespace onnxruntime
|