Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
80203608
Commit
80203608
authored
Nov 13, 2018
by
Paul
Browse files
Merge branch 'multi_broadcast-fix'
parents
3ef07c55
c74767b1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
1 deletion
+5
-1
src/onnx/onnx.cpp
src/onnx/onnx.cpp
+5
-1
No files found.
src/onnx/onnx.cpp
View file @
80203608
...
@@ -113,7 +113,7 @@ struct onnx_parser
...
@@ -113,7 +113,7 @@ struct onnx_parser
}
}
return
prog
.
add_instruction
(
x
,
args
);
return
prog
.
add_instruction
(
x
,
args
);
}
}
else
else
if
(
args
[
0
]
->
get_shape
()
!=
args
[
1
]
->
get_shape
())
{
{
// Example:
// Example:
// s0 = (3,2,4,5) and s1 = (2,1,1)
// s0 = (3,2,4,5) and s1 = (2,1,1)
...
@@ -149,6 +149,10 @@ struct onnx_parser
...
@@ -149,6 +149,10 @@ struct onnx_parser
auto
l1
=
prog
.
add_instruction
(
op
::
multibroadcast
{
output_lens
},
args
[
1
]);
auto
l1
=
prog
.
add_instruction
(
op
::
multibroadcast
{
output_lens
},
args
[
1
]);
return
prog
.
add_instruction
(
x
,
l0
,
l1
);
return
prog
.
add_instruction
(
x
,
l0
,
l1
);
}
}
else
{
return
prog
.
add_instruction
(
x
,
args
);
}
});
});
}
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment