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
ee55f2c8
Commit
ee55f2c8
authored
Jan 29, 2019
by
Khalique
Browse files
formatting
parent
8bf2940a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
src/tf/tf.cpp
src/tf/tf.cpp
+3
-2
No files found.
src/tf/tf.cpp
View file @
ee55f2c8
...
@@ -159,7 +159,8 @@ struct tf_parser
...
@@ -159,7 +159,8 @@ struct tf_parser
std
::
size_t
axis
=
args
[
axis_idx
]
->
eval
().
at
<
int64_t
>
();
std
::
size_t
axis
=
args
[
axis_idx
]
->
eval
().
at
<
int64_t
>
();
op
::
concat
op
{
axis
};
op
::
concat
op
{
axis
};
// return only first N arguments (assuming last index is the axis value)
// return only first N arguments (assuming last index is the axis value)
return
prog
.
add_instruction
(
op
,
std
::
vector
<
instruction_ref
>
(
args
.
begin
(),
args
.
begin
()
+
axis
));
return
prog
.
add_instruction
(
op
,
std
::
vector
<
instruction_ref
>
(
args
.
begin
(),
args
.
begin
()
+
axis
));
}
}
instruction_ref
parse_constant
(
const
std
::
string
&
,
instruction_ref
parse_constant
(
const
std
::
string
&
,
...
@@ -236,7 +237,7 @@ struct tf_parser
...
@@ -236,7 +237,7 @@ struct tf_parser
}
}
}
}
auto
l0
=
args
[
0
];
auto
l0
=
args
[
0
];
if
(
l0
->
name
()
==
"@param"
)
if
(
l0
->
name
()
==
"@param"
)
{
{
if
(
is_nhwc
)
if
(
is_nhwc
)
l0
=
prog
.
add_instruction
(
op
::
transpose
{{
0
,
3
,
1
,
2
}},
l0
);
l0
=
prog
.
add_instruction
(
op
::
transpose
{{
0
,
3
,
1
,
2
}},
l0
);
...
...
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