mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-22 19:23:30 +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.
31 lines
822 B
C++
31 lines
822 B
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include "core/optimizer/rewrite_rule.h"
|
|
|
|
namespace onnxruntime {
|
|
|
|
/**
|
|
@Class ConvAddFusion
|
|
|
|
Rewrite rule that fuses two Conv+Add nodes to a single Conv node.
|
|
|
|
It is attempted to be triggered only on nodes with op type "Conv".
|
|
*/
|
|
class ConvAddFusion : public RewriteRule {
|
|
public:
|
|
ConvAddFusion() noexcept : RewriteRule("ConvAddFusion") {}
|
|
|
|
std::vector<std::string> TargetOpTypes() const noexcept override {
|
|
return {"Conv"};
|
|
}
|
|
|
|
private:
|
|
bool SatisfyCondition(const Graph& graph, const Node& node, const logging::Logger& logger) const override;
|
|
|
|
Status Apply(Graph& graph, Node& node, RewriteRuleEffect& rule_effect, const logging::Logger& logger) const override;
|
|
};
|
|
|
|
} // namespace onnxruntime
|