saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-13 18:08:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

changed code per reviewer

Browse source
This commit is contained in:
Chi Lo 2025-02-04 14:39:57 -08:00
parent 4c2697c687
commit 2b81789507
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
onnxruntime/core/optimizer/constant_folding.h
View file

@ -28,15 +28,19 @@ class ConstantFolding : public GraphTransformer {
const InlinedHashSet<std::string_view>& compatible_execution_providers = {},
const InlinedHashSet<std::string>& excluded_initializers = {}) noexcept;
/* Same as above but with a name provided by derived class.
*/
protected:
/**
* Same as the constructor above but with a name provided by derived class.
*/
ConstantFolding(const std::string& name,
const IExecutionProvider& execution_provider,
bool skip_dequantize_linear,
const ConfigOptions& config_options,
const InlinedHashSet<std::string_view>& compatible_execution_providers = {},
const InlinedHashSet<std::string>& excluded_initializers = {}) noexcept;
/**
* Derived class can implement this virtual function to limit the nodes that can be constant folded.
*/
virtual bool AllowConstantFolding(const Node& node) const { return true; }
private: