onnxruntime
Fix Transpose, Cast and MatMul fusion when Cast and/or Fusion feeds multiple nodes.
#7021
Merged

Fix Transpose, Cast and MatMul fusion when Cast and/or Fusion feeds multiple nodes. #7021

satyajandhyala
satyajandhyala satyajandhyala added component:ortmodule
satyajandhyala satyajandhyala force pushed from b85d9426 to 98db125e 4 years ago
satyajandhyala satyajandhyala changed the title [WIP] Fix:Transpose, Cast and MatMul fusion when Cast feeds multiple nodes. Fix Transpose, Cast and MatMul fusion when Cast and/or Fusion feeds multiple nodes. 4 years ago
satyajandhyala satyajandhyala marked this pull request as ready for review 4 years ago
satyajandhyala satyajandhyala requested a review 4 years ago
ytaous
ytaous commented on 2021-03-17
ytaous
ytaous commented on 2021-03-17
ytaous
ytaous
ytaous commented on 2021-03-17
satyajandhyala
satyajandhyala satyajandhyala force pushed from 48fb2677 to 6ffca393 4 years ago
satyajandhyala satyajandhyala force pushed from 6ffca393 to cde69871 4 years ago
satyajandhyala satyajandhyala changed the base branch from thiagofc/ortmodule-api to master 4 years ago
satyajandhyala satyajandhyala requested a review from baijumeswani baijumeswani 4 years ago
satyajandhyala satyajandhyala requested a review from BowenBao BowenBao 4 years ago
satyajandhyala satyajandhyala requested a review from liqunfu liqunfu 4 years ago
satyajandhyala satyajandhyala requested a review from spandantiwari spandantiwari 4 years ago
satyajandhyala satyajandhyala requested a review from thiagocrepaldi thiagocrepaldi 4 years ago
satyajandhyala satyajandhyala changed the base branch from master to thiagofc/ortmodule-api 4 years ago
satyajandhyala satyajandhyala force pushed from cde69871 to 744e0fe0 4 years ago
satyajandhyala satyajandhyala changed the base branch from thiagofc/ortmodule-api to master 4 years ago
satyajandhyala Added code to generate new models for different scenarios.
56764b6d
satyajandhyala Added new Transpose+Cast+MatMul => Cast+FusedMatMul test scenarios.
74840b24
satyajandhyala The Cast node may feed more than one node.
1450b7fc
satyajandhyala Perform Transpose, Cast and MatMul fusion on the both inputs in the s…
3406a7a2
satyajandhyala Fixed the test model structure as intended and added variable names t…
c0b55929
satyajandhyala Change input/output tensor dimentions to make MalMul valid; updated …
8ec73f29
satyajandhyala Added new ONNX test input files.
62b18a5e
satyajandhyala Transpose node may feed multiple nodes and still may be fused with Ma…
a5a67b6b
satyajandhyala Added a new test case.
618881eb
satyajandhyala Used remove_nodes to keep track of the nodes to remove
b2b59e06
satyajandhyala Limit columns to 120
ecc609cf
satyajandhyala Removed saving model
0259837e
satyajandhyala Renamed GetTransposeNodeFromCast to ReorderCastAndTranspose
994022c1
satyajandhyala Use original op count to verify post transformation op count.
5430d92e
satyajandhyala satyajandhyala force pushed from 744e0fe0 to 5430d92e 4 years ago
satyajandhyala satyajandhyala requested a review from ytaous ytaous 4 years ago
ytaous
ytaous approved these changes on 2021-03-18
satyajandhyala satyajandhyala merged 8bc275e9 into master 4 years ago
satyajandhyala satyajandhyala deleted the ortmodule/transpose_cast_matmul_fusion branch 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone