Public
  1. Public

onnxruntime

Public

Commits

Bowen BaoGitHub
dce76c15e7c
Bowen Bao authored and GitHub committed a776b57160e
Add shape inference to custom symbolic functions (#7937)

**Description**: As title.

**Motivation and Context**
- PyTorch ONNX exporter heavily depends on ONNX shape inference to export accurate and efficient model. Custom symbolic function exports the op as contrib ops, thus exporter is unable to perform standard onnx shape inference. Models with dynamic shape inputs are affected.