microsoft/onnxscript

Constant-Folding Registry

Open

#2,507 opened on Aug 22, 2025

View on GitHub
 (2 comments) (0 reactions) (0 assignees)Python (117 forks)auto 404
category: questioncontribution welcomegood first issuemodule: optimizer

Repository metrics

Stars
 (445 stars)
PR merge metrics
 (PR metrics pending)

Description

Authoring custom constant-folding for operators via the register method in onnxscript.optimizer._constant_folding feels quite convenient (see #2506 for example) - are there any plans to properly expose this to the user to make constant folding customizable? I mean, it is already possible to simply from onnxscript.optimizer._constant_folding import register, but this is importing from a protected memeber (at least the linter complains), so I guess it is not meant to be used this way, at least for now?

Contributor guide