onnxruntime/include/onnxruntime/core
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
..
common
eager
framework
graph CoreML: Aggregated changes to add all required ops for priority model (#21472) 2024-07-26 08:29:33 +10:00
optimizer
platform
providers
session Add support tensor element type for register custom op shape infer function (#21387) 2024-07-29 09:45:52 -07:00