Skip to content

Commit

Permalink
[Relay][OP] Gather_nd exposed to relay (#2945)
Browse files Browse the repository at this point in the history
* gather_nd added

* gather_nd test added

* more test added

* fix lint

* fix build error

* fix lint

* comments addressed
  • Loading branch information
Laurawly authored and ajtulloch committed Apr 5, 2019
1 parent 6537bdb commit 5217933
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions tests/python/frontend/mxnet/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -505,12 +505,6 @@ def verify(xshape, yshape, y_data):
verify((2, 2), (2, 3), [[1, 1, 0], [0, 1, 0]])
verify((2, 2, 2), (2, 2), [[0, 1], [1, 0]])

def test_forward_bilinear_resize():
# add tests including scale_height and scale_width when mxnet is updated to version 1.5
data = mx.sym.var('data')
mx_sym = mx.sym.contrib.BilinearResize2D(data, height=5, width=10)
verify_mxnet_frontend_impl(mx_sym, (1, 2, 3, 4), (1, 2, 5, 10))


if __name__ == '__main__':
test_forward_mlp()
Expand Down Expand Up @@ -549,4 +543,3 @@ def test_forward_bilinear_resize():
test_forward_smooth_l1()
test_forward_take()
test_forward_gather_nd()
test_forward_bilinear_resize()

0 comments on commit 5217933

Please sign in to comment.