New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Custom op shape inference API #17737

Merged

RandySheriffH merged 17 commits into main from rashuai/CustomShapeInfer

Oct 13, 2023

Contributor

RandySheriffH commented Sep 29, 2023 •

edited

Loading

Add c/cxx API to allow custom ops do shape inference.

RandyShuai added 5 commits

September 27, 2023 17:30


          implement c api

cb5cdc6


          extend schema

b784379


          make shape infer static

7ea7307


          add ut

e594dda


          merge main

88a453c

RandySheriffH changed the title ~~[WIP] Custom op shape inference API~~ Custom op shape inference API


          init new callback

42b4b0a

RandySheriffH marked this pull request as ready for review

October 3, 2023 20:35

RandySheriffH commented

View reviewed changes

include/onnxruntime/core/session/onnxruntime_lite_custom_op.h Outdated Show resolved Hide resolved

RandySheriffH commented

View reviewed changes

include/onnxruntime/core/session/onnxruntime_lite_custom_op.h Outdated Show resolved Hide resolved

RandySheriffH commented

View reviewed changes

include/onnxruntime/core/session/onnxruntime_lite_custom_op.h Outdated Show resolved Hide resolved

RandySheriffH commented

View reviewed changes

include/onnxruntime/core/session/onnxruntime_lite_custom_op.h Outdated Show resolved Hide resolved

RandySheriffH commented

View reviewed changes

onnxruntime/core/graph/graph.cc Outdated Show resolved Hide resolved

RandySheriffH commented

View reviewed changes

onnxruntime/core/graph/graph.cc Outdated Show resolved Hide resolved


          support attr from ctx

497dfbe

RandySheriffH commented

View reviewed changes

onnxruntime/core/session/standalone_op_invoker.cc Outdated Show resolved Hide resolved

RandySheriffH commented

View reviewed changes

onnxruntime/core/session/standalone_op_invoker.cc Outdated Show resolved Hide resolved


          add testing model

baa267b

RandySheriffH commented

View reviewed changes

onnxruntime/test/testdata/custom_op_library/cpu/cpu_ops.cc Show resolved Hide resolved

RandySheriffH and others added 2 commits

October 5, 2023 18:04


          format code

99c84d7


          drop templates

86f9ca7

souptc reviewed

View reviewed changes

include/onnxruntime/core/session/onnxruntime_c_api.h Show resolved Hide resolved

souptc reviewed

View reviewed changes

include/onnxruntime/core/session/onnxruntime_cxx_inline.h Outdated Show resolved Hide resolved

souptc reviewed

View reviewed changes

include/onnxruntime/core/session/onnxruntime_cxx_inline.h Outdated Show resolved Hide resolved

RandyShuai added 6 commits

October 6, 2023 14:06


          fix comments

9b9d9a9


          fix typo

96d5e97


          address comments

5a9665f


          enrich symbolic ut

ab1526c


          set default dim

c0b89e1


          tune condition

f13b858

souptc reviewed

View reviewed changes

include/onnxruntime/core/session/onnxruntime_cxx_api.h Show resolved Hide resolved

souptc previously approved these changes

View reviewed changes

Member

souptc left a comment


          add doc string to shape infer context

4f12f23

RandySheriffH dismissed souptc’s stale review via

4f12f23

October 13, 2023 03:10

souptc approved these changes

View reviewed changes

RandySheriffH merged commit c6c3555 into main

91 checks passed

RandySheriffH deleted the rashuai/CustomShapeInfer branch

October 13, 2023 19:57

jchen351 pushed a commit that referenced this pull request


          Custom op shape inference API (#17737)

7b86b65

Add c/cxx API to allow custom ops do shape  inference.

---------

Co-authored-by: Randy Shuai <[email protected]>

kleiti pushed a commit to kleiti/onnxruntime that referenced this pull request


          Custom op shape inference API (microsoft#17737)

2e62e69

Add c/cxx API to allow custom ops do shape  inference.

---------

Co-authored-by: Randy Shuai <[email protected]>

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet