Skip to content

Commit

Permalink
fix pylint
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed May 20, 2021
1 parent 4be279d commit 76dce71
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/tvm/relay/frontend/onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -1413,7 +1413,8 @@ def _impl_v1(cls, inputs, attr, params):
class GatherND(OnnxOpConverter):
"""Operator converter for GatherND."""

def _impl_common(data, indices, batch_dims=0):
@classmethod
def _impl_common(cls, data, indices, batch_dims=0):
indices_dims = len(infer_shape(indices))
indices = _op.transpose(indices, axes=[-1] + list(range(indices_dims - 1)))
return _op.gather_nd(data, indices, batch_dims)
Expand Down

0 comments on commit 76dce71

Please sign in to comment.