Fix Transpose, Cast and MatMul fusion when Cast and/or Fusion feeds multiple nodes. #7021
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
marked this pull request as ready for review 4 years ago
ytaous
commented
on 2021-03-17
ytaous
commented
on 2021-03-17
ytaous
commented
on 2021-03-17
satyajandhyala
changed the base branch from
thiagofc/ortmodule-api
to
master
4 years ago
satyajandhyala
changed the base branch from
master
to
thiagofc/ortmodule-api
4 years ago
satyajandhyala
changed the base branch from
thiagofc/ortmodule-api
to
master
4 years ago
Added code to generate new models for different scenarios.
56764b6d
Added new Transpose+Cast+MatMul => Cast+FusedMatMul test scenarios.
74840b24
The Cast node may feed more than one node.
1450b7fc
Perform Transpose, Cast and MatMul fusion on the both inputs in the s…
3406a7a2
Fixed the test model structure as intended and added variable names t…
c0b55929
Change input/output tensor dimentions to make MalMul valid; updated …
8ec73f29
Added new ONNX test input files.
62b18a5e
Transpose node may feed multiple nodes and still may be fused with Ma…
a5a67b6b
Added a new test case.
618881eb
Used remove_nodes to keep track of the nodes to remove
b2b59e06
Limit columns to 120
ecc609cf
Removed saving model
0259837e
Renamed GetTransposeNodeFromCast to ReorderCastAndTranspose
994022c1
Use original op count to verify post transformation op count.
5430d92e
ytaous
approved these changes
on 2021-03-18
satyajandhyala
deleted the ortmodule/transpose_cast_matmul_fusion branch 4 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub