mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-23 19:32:23 +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>
17 lines
485 B
C++
17 lines
485 B
C++
#pragma once
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "onnx/onnx_pb.h"
|
|
|
|
namespace onnxruntime {
|
|
|
|
struct FunctionTemplate {
|
|
// The generated schema for the function proto
|
|
// ORT rely on it to run type/shape inference
|
|
std::unique_ptr<ONNX_NAMESPACE::OpSchema> op_schema_;
|
|
// reference to the function proto in local function
|
|
const ONNX_NAMESPACE::FunctionProto* onnx_func_proto_;
|
|
};
|
|
|
|
} // namespace onnxruntime
|