Add `ShouldConvertDataLayoutForOp()` API to allow EPs to customize layout sensitive ops #25147
save work
4440e7d6
update JS EP
543621e1
update webgpu EP
59a590a0
update CUDA EP
cf2ad94f
update QNN EP
0396b630
fix test
9b0e1ad4
add GridSample contrib op to ORT layout sensitive ops
bc18af03
handle contrib GridSample in CUDA EP too
82deccf9
edgchen1
marked this pull request as ready for review 1 year ago
Merge remote-tracking branch 'origin/main' into edgchen1/ep_layout_se…
6e9b240a
Merge remote-tracking branch 'origin/main' into edgchen1/ep_layout_se…
e5429c92
Add ORT EP API.
594bd5cb
format, convert string_view to string
bc931c24
fix issues
85e01030
edgchen1
changed the title Add `IExecutionProvider::ShouldConvertNodeLayoutToNhwc()` to allow EPs to customize layout sensitive ops Add `ShouldConvertNodeLayoutToNhwc()` API to allow EPs to customize layout sensitive ops 1 year ago
update comment
1abe81a8
add parameter for target data layout
1ff6c221
fix build
1362d6f2
add layout check to QNN EP's impl
e1453fa9
rename ShouldConvertNodeLayout to ShouldConvertDataLayoutForOp
c94ee528
edgchen1
changed the title Add `ShouldConvertNodeLayoutToNhwc()` API to allow EPs to customize layout sensitive ops Add `ShouldConvertDataLayoutForOp()` API to allow EPs to customize layout sensitive ops 1 year ago
add DataLayout::Default as alias for NCHW
900891d8
edgchen1
merged
6346cdd6
into main 1 year ago
edgchen1
deleted the edgchen1/ep_layout_sensitive_ops branch 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub