Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
T
tic
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
wenyuanbo
tic
Commits
ad7ffd35
Commit
ad7ffd35
authored
7 years ago
by
Joshua Z. Zhang
Committed by
Tianqi Chen
6 years ago
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add softmaxoutput (#207)
parent
8b6f80c2
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
0 deletions
+7
-0
nnvm/python/nnvm/frontend/mxnet.py
+7
-0
No files found.
nnvm/python/nnvm/frontend/mxnet.py
View file @
ad7ffd35
...
...
@@ -184,6 +184,12 @@ def _split(inputs, attrs):
new_attrs
[
'axis'
]
=
attrs
.
get
(
'axis'
,
1
)
return
_get_nnvm_op
(
op_name
)(
*
inputs
,
**
new_attrs
)
def
_softmax_output
(
inputs
,
attrs
):
op_name
,
new_attrs
=
'softmax'
,
{}
if
_parse_bool_str
(
attrs
,
'multi_output'
):
new_attrs
[
'axis'
]
=
1
return
_get_nnvm_op
(
op_name
)(
inputs
[
0
],
**
new_attrs
)
_identity_list
=
[
'__add_scalar__'
,
'__add_symbol__'
,
'__div_scalar__'
,
'__div_symbol__'
,
'__mul_scalar__'
,
'__mul_symbol__'
,
'__pow_scalar__'
,
'__rdiv_scalar__'
,
'__rpow_scalar__'
,
...
...
@@ -217,6 +223,7 @@ _convert_map = {
'Pooling_v1'
:
_pooling
,
'Reshape'
:
_reshape
,
'Softmax'
:
_rename
(
'softmax'
),
'SoftmaxOutput'
:
_softmax_output
,
'concat'
:
_concat
,
'max_axis'
:
_rename
(
'max'
),
'min_axis'
:
_rename
(
'min'
),
...
...
This diff is collapsed.
Click to expand it.
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