saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix EP in transform (#8909)

Browse source
This commit is contained in:
ashbhandare 2021-08-31 13:52:57 -07:00 committed by GitHub
parent dc75a135c8
commit cd4b9f7753
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/optimizer/matmul_transpose_fusion.cc
View file

@ -142,13 +142,14 @@ static Node* ReorderCastAndTranspose(Graph& graph, Node* cast,
const std::vector<NodeArg*> new_transpose_input_defs = {&new_cast_output};
const std::vector<NodeArg*> new_transpose_output_defs = {cast_output};
(void)graph.AddNode(graph.GenerateNodeName(cast->Name() + "_transformed"),
Node& new_cast = graph.AddNode(graph.GenerateNodeName(cast->Name() + "_transformed"),
cast->OpType(),
"Created a new Cast node to interchange Cast and Transpose nodes",
new_cast_input_defs,
new_cast_output_defs,
&cast->GetAttributes(),
cast->Domain());
new_cast.SetExecutionProviderType(cast->GetExecutionProviderType());
Node& new_transpose = graph.AddNode(graph.GenerateNodeName(transpose->Name() + "_transformed"),
transpose->OpType(),
@ -157,6 +158,7 @@ static Node* ReorderCastAndTranspose(Graph& graph, Node* cast,
new_transpose_output_defs,
&transpose->GetAttributes(),
transpose->Domain());
new_transpose.SetExecutionProviderType(transpose->GetExecutionProviderType());
size_t consumers = UpdateConsumerCount(graph, transpose->MutableOutputDefs()[0], consumer_count);
graph_utils::RemoveNodeOutputEdges(graph, *cast);