onnxruntime/include/onnxruntime/core/session
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
..
environment.h ExecutionProvider API refactor - move allocator from EP level to SessionState level and indexed by OrtDevice (#15833) 2023-06-19 17:44:45 -07:00
experimental_onnxruntime_cxx_api.h
experimental_onnxruntime_cxx_inline.h
onnxruntime_c_api.h Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
onnxruntime_cxx_api.h enable model with external data be loaded from memory buffer (#19089) 2024-04-17 19:01:01 -07:00
onnxruntime_cxx_inline.h [Fix] C++ API SetOutputShape for register custom op. (#21366) 2024-07-23 16:51:00 -07:00
onnxruntime_float16.h Work on eliminating Internal Compiler Error (#16741) 2023-07-18 10:17:52 -07:00
onnxruntime_lite_custom_op.h Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
onnxruntime_run_options_config_keys.h cuda graph enhancement (#19636) 2024-03-07 10:15:18 -08:00
onnxruntime_session_options_config_keys.h add transform part of the dq matmul tool chain (#21374) 2024-07-19 22:55:15 -07:00
snippets.dox