Public
  1. Public

onnxruntime

Public

Commits

mingyueliuhGitHub
94eb70d9834
mingyueliuh authored and GitHub committed d8888136e3c
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>