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
eef9de6f
"vscode:/vscode.git/clone" did not exist on "cff1144da0f0c2b835cf9fe623c41b1004a599ed"
Commit
eef9de6f
authored
Aug 21, 2018
by
Paul
Browse files
Formatting
parent
ea87f25e
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
11 additions
and
8 deletions
+11
-8
src/include/migraph/operators.hpp
src/include/migraph/operators.hpp
+1
-1
src/onnx/verify_onnx.cpp
src/onnx/verify_onnx.cpp
+3
-3
test/gpu/miopen.cpp
test/gpu/miopen.cpp
+7
-4
No files found.
src/include/migraph/operators.hpp
View file @
eef9de6f
src/onnx/verify_onnx.cpp
View file @
eef9de6f
test/gpu/miopen.cpp
View file @
eef9de6f
...
...
@@ -122,7 +122,9 @@ migraph::argument run_gpu()
return
migraph
::
gpu
::
from_gpu
(
p
.
eval
(
m
));
}
void
verify_args
(
const
std
::
string
&
name
,
const
migraph
::
argument
&
cpu_arg
,
const
migraph
::
argument
&
gpu_arg
)
void
verify_args
(
const
std
::
string
&
name
,
const
migraph
::
argument
&
cpu_arg
,
const
migraph
::
argument
&
gpu_arg
)
{
visit_all
(
cpu_arg
,
gpu_arg
)([
&
](
auto
cpu
,
auto
gpu
)
{
if
(
not
migraph
::
verify_range
(
cpu
,
gpu
))
...
...
@@ -391,7 +393,8 @@ struct test_conv_bn_relu_pooling
auto
variance
=
p
.
add_literal
(
migraph
::
abs
(
migraph
::
generate_literal
(
vars
,
1
)));
auto
scale
=
p
.
add_literal
(
migraph
::
abs
(
migraph
::
generate_literal
(
vars
,
2
)));
auto
bias
=
p
.
add_literal
(
migraph
::
abs
(
migraph
::
generate_literal
(
vars
,
3
)));
auto
bn
=
p
.
add_instruction
(
migraph
::
batch_norm_inference
{},
conv
,
mean
,
variance
,
scale
,
bias
);
auto
bn
=
p
.
add_instruction
(
migraph
::
batch_norm_inference
{},
conv
,
mean
,
variance
,
scale
,
bias
);
auto
relu
=
p
.
add_instruction
(
migraph
::
activation
{
"relu"
},
bn
);
p
.
add_instruction
(
migraph
::
pooling
{
"average"
,
{
1
,
1
},
{
2
,
2
},
{
3
,
3
}},
relu
);
return
p
;
...
...
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