I have trained a model and export to onnx, but I got an error when infer by ort:
Non-zero status code returned while running Expand node. Name:'/GaussianDiffusionModel/Expand' Status Message: /GaussianDiffusionModel/Expand: left operand cannot broadcast on dim 0 LeftShape: {110}, RightShape: {118}
when I unable the enable_mem_reuse option in python, it works, but I cannot find the modification method in cpp, how can I unable this option with c/cpp program and resolve this problem?
Describe the issue
I have trained a model and export to onnx, but I got an error when infer by ort: Non-zero status code returned while running Expand node. Name:'/GaussianDiffusionModel/Expand' Status Message: /GaussianDiffusionModel/Expand: left operand cannot broadcast on dim 0 LeftShape: {110}, RightShape: {118}
when I unable the enable_mem_reuse option in python, it works, but I cannot find the modification method in cpp, how can I unable this option with c/cpp program and resolve this problem?
To reproduce
Urgency
No response
Platform
Linux
OS Version
CentOS
ONNX Runtime Installation
Released Package
ONNX Runtime Version or Commit ID
1.18.1
ONNX Runtime API
C++
Architecture
X86
Execution Provider
Default CPU, CUDA
Execution Provider Library Version
CUDA 11.8