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
98338dea
Commit
98338dea
authored
Apr 12, 2019
by
Shucai Xiao
Browse files
refine the print out information
parent
2093d79e
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
8 additions
and
7 deletions
+8
-7
src/include/migraphx/op/gru.hpp
src/include/migraphx/op/gru.hpp
+2
-2
src/include/migraphx/op/lstm.hpp
src/include/migraphx/op/lstm.hpp
+2
-2
src/include/migraphx/op/rnn.hpp
src/include/migraphx/op/rnn.hpp
+2
-2
src/rewrite_rnn.cpp
src/rewrite_rnn.cpp
+2
-1
No files found.
src/include/migraphx/op/gru.hpp
View file @
98338dea
...
@@ -30,8 +30,8 @@ struct gru
...
@@ -30,8 +30,8 @@ struct gru
template
<
class
Self
,
class
F
>
template
<
class
Self
,
class
F
>
static
auto
reflect
(
Self
&
self
,
F
f
)
static
auto
reflect
(
Self
&
self
,
F
f
)
{
{
return
pack
(
f
(
self
.
hidden_size
,
"hidden
size"
),
return
pack
(
f
(
self
.
hidden_size
,
"hidden
_
size"
),
f
(
self
.
actv_funcs
,
"act
ivation functions
"
),
f
(
self
.
actv_funcs
,
"act
v_func
"
),
f
(
self
.
direction
,
"direction"
),
f
(
self
.
direction
,
"direction"
),
f
(
self
.
clip
,
"clip"
),
f
(
self
.
clip
,
"clip"
),
f
(
self
.
linear_before_reset
,
"linear_before_reset"
));
f
(
self
.
linear_before_reset
,
"linear_before_reset"
));
...
...
src/include/migraphx/op/lstm.hpp
View file @
98338dea
...
@@ -28,8 +28,8 @@ struct lstm
...
@@ -28,8 +28,8 @@ struct lstm
template
<
class
Self
,
class
F
>
template
<
class
Self
,
class
F
>
static
auto
reflect
(
Self
&
self
,
F
f
)
static
auto
reflect
(
Self
&
self
,
F
f
)
{
{
return
pack
(
f
(
self
.
hidden_size
,
"hidden
size"
),
return
pack
(
f
(
self
.
hidden_size
,
"hidden
_
size"
),
f
(
self
.
actv_funcs
,
"act
ivation functions
"
),
f
(
self
.
actv_funcs
,
"act
v_func
"
),
f
(
self
.
direction
,
"direction"
),
f
(
self
.
direction
,
"direction"
),
f
(
self
.
input_forget
,
"input_forget"
));
f
(
self
.
input_forget
,
"input_forget"
));
}
}
...
...
src/include/migraphx/op/rnn.hpp
View file @
98338dea
...
@@ -28,8 +28,8 @@ struct rnn
...
@@ -28,8 +28,8 @@ struct rnn
template
<
class
Self
,
class
F
>
template
<
class
Self
,
class
F
>
static
auto
reflect
(
Self
&
self
,
F
f
)
static
auto
reflect
(
Self
&
self
,
F
f
)
{
{
return
pack
(
f
(
self
.
hidden_size
,
"hidden
size"
),
return
pack
(
f
(
self
.
hidden_size
,
"hidden
_
size"
),
f
(
self
.
actv_funcs
,
"act
ivation functions
"
),
f
(
self
.
actv_funcs
,
"act
v_func
"
),
f
(
self
.
direction
,
"direction"
),
f
(
self
.
direction
,
"direction"
),
f
(
self
.
clip
,
"clip"
));
f
(
self
.
clip
,
"clip"
));
}
}
...
...
src/rewrite_rnn.cpp
View file @
98338dea
...
@@ -1170,7 +1170,8 @@ std::vector<operation> rewrite_rnn::lstm_actv_funcs(instruction_ref ins) const
...
@@ -1170,7 +1170,8 @@ std::vector<operation> rewrite_rnn::lstm_actv_funcs(instruction_ref ins) const
namespace
op
{
namespace
op
{
std
::
ostream
&
operator
<<
(
std
::
ostream
&
os
,
rnn_direction
v
)
std
::
ostream
&
operator
<<
(
std
::
ostream
&
os
,
rnn_direction
v
)
{
{
os
<<
static_cast
<
std
::
underlying_type
<
rnn_direction
>::
type
>
(
v
);
std
::
vector
<
std
::
string
>
rnn_direction_str
=
{
"forward"
,
"reverse"
,
"bidirectional"
};
os
<<
rnn_direction_str
[
static_cast
<
std
::
underlying_type
<
rnn_direction
>::
type
>
(
v
)];
return
os
;
return
os
;
}
}
}
// namespace op
}
// namespace op
...
...
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