mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-25 22:26:24 +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>
31 lines
679 B
Objective-C
31 lines
679 B
Objective-C
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include "core/common/common.h"
|
|
#include "core/graph/indexed_sub_graph.h"
|
|
|
|
namespace onnxruntime {
|
|
class Graph;
|
|
class Node;
|
|
} // namespace onnxruntime
|
|
|
|
namespace onnxruntime {
|
|
|
|
/**
|
|
@class Function
|
|
Class representing a Function.
|
|
*/
|
|
class Function {
|
|
public:
|
|
virtual ~Function() = default;
|
|
|
|
/** Gets the Graph instance for the Function body subgraph. */
|
|
virtual const onnxruntime::Graph& Body() const = 0;
|
|
|
|
/** Gets the Mutable Graph instance for the Function body subgraph. */
|
|
virtual onnxruntime::Graph& MutableBody() = 0;
|
|
};
|
|
|
|
} // namespace onnxruntime
|