mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-22 22:01:08 +00:00
109b3cb450
1. Use the session logger if it is available. 2. Don't disable warning 4100 globally. We should fix the warnings instead of disabling it.
46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
#include "core/common/logging/logging.h"
|
|
#include "core/graph/function.h"
|
|
#include "core/graph/model.h"
|
|
|
|
namespace onnxruntime {
|
|
class Graph;
|
|
class Node;
|
|
} // namespace onnxruntime
|
|
|
|
namespace onnxruntime {
|
|
|
|
// Function representation class.
|
|
class FunctionImpl final : public Function {
|
|
public:
|
|
FunctionImpl(const onnxruntime::Graph& graph,
|
|
std::unique_ptr<IndexedSubGraph> customized_func,
|
|
const logging::Logger& logger);
|
|
|
|
FunctionImpl(const onnxruntime::Graph& graph,
|
|
const onnxruntime::NodeIndex& node_index,
|
|
const ONNX_NAMESPACE::FunctionProto& onnx_func,
|
|
const logging::Logger& logger);
|
|
|
|
~FunctionImpl() override;
|
|
|
|
const ONNX_NAMESPACE::OpSchema& OpSchema() const override;
|
|
|
|
const onnxruntime::Graph& Body() const override;
|
|
|
|
const IndexedSubGraph& GetIndexedSubGraph() const override;
|
|
|
|
const ONNX_NAMESPACE::FunctionProto* GetFuncProto() const;
|
|
|
|
private:
|
|
const onnxruntime::Graph* const parent_graph_;
|
|
std::unique_ptr<IndexedSubGraph> customized_func_body_;
|
|
std::unique_ptr<ONNX_NAMESPACE::OpSchema> op_schema_;
|
|
onnxruntime::Model body_;
|
|
ONNX_NAMESPACE::FunctionProto onnx_func_proto_;
|
|
};
|
|
|
|
} // namespace onnxruntime
|