Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Fix a comment and rebase
Browse files Browse the repository at this point in the history
  • Loading branch information
vandanavk committed Aug 9, 2019
1 parent 387e909 commit def50eb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/mxnet/contrib/onnx/mx2onnx/_op_translations.py
Original file line number Diff line number Diff line change
Expand Up @@ -2051,7 +2051,7 @@ def convert_broadcast_to(node, **kwargs):

@mx_op.register("topk")
def convert_topk(node, **kwargs):
"""Map MXNet's size_array operator attributes to onnx's Size operator
"""Map MXNet's topk operator attributes to onnx's TopK operator
and return the created node.
"""
name, input_nodes, attrs = get_inputs(node, kwargs)
Expand Down

0 comments on commit def50eb

Please sign in to comment.