onnxruntime/onnxruntime
mingyueliuh d8888136e3
Add support tensor element type for register custom op shape infer function (#21387)
### Description
Functionality extension for the SetOutputShape method in custom op shape inference.


### Motivation and Context
-  **SetOutputShape** Interface enhancement Actually, the shape infer function need set the tensor type and shape ,Add a parameter **type** to allow users to specify the tensor type, and set **ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT** as default value to ensure compatibility.

Co-authored-by: mingyue <mingyue@amd.com>
2024-07-29 09:45:52 -07:00
..
contrib_ops Security fuzz address sanitizer fix Bug #2 and #3 (#21528) 2024-07-27 11:10:52 -07:00
core Add support tensor element type for register custom op shape infer function (#21387) 2024-07-29 09:45:52 -07:00
python CoreML: Aggregated changes to add all required ops for priority model (#21472) 2024-07-26 08:29:33 +10:00
test pick changes from https://github.com/onnx/onnx/pull/6195 to fix heap-buffer-overflow in onnx::convPoolShapeInference (#21507) 2024-07-27 15:58:36 -07:00
tool/etw Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
wasm Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
__init__.py
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings