onnxruntime/include/onnxruntime/core
mingyueliuh 86cedc6832
[Fix] C++ API SetOutputShape for register custom op. (#21366)
### Description
Bug fix  for the SetOutputShape method in custom op shape inference.


### Motivation and Context

- Bug a :   A obvious bug that will cause all dimensions to be 1. 

https://github.com/microsoft/onnxruntime/blob/main/include/onnxruntime/core/session/onnxruntime_cxx_inline.h#L2014
integer_dims.push_back(dim.IsInt()); ->
integer_dims.push_back(dim.AsInt());

- Bug b :  vector out of range error
op's input maybe a scalar and shape is empty.

https://github.com/microsoft/onnxruntime/blob/main/include/onnxruntime/core/session/onnxruntime_cxx_inline.h#L1985

---------

Co-authored-by: mingyue <mingyue@amd.com>
2024-07-23 16:51:00 -07:00
..
common Remove core/common/gsl.h (#20894) 2024-07-08 18:09:39 -07:00
eager Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
framework Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
graph VitisAI EP Context Model (#20926) 2024-07-12 21:22:58 -07:00
optimizer add transform part of the dq matmul tool chain (#21374) 2024-07-19 22:55:15 -07:00
platform Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
providers Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
session [Fix] C++ API SetOutputShape for register custom op. (#21366) 2024-07-23 16:51:00 -07:00