Skip to content

Commit 753f876

Browse files
committed
cleanup (apache#21)
1 parent 9b982c3 commit 753f876

File tree

2 files changed

+0
-15
lines changed

2 files changed

+0
-15
lines changed

nnvm/example/src/operator.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,6 @@ NNVM_REGISTER_OP(__add_symbol__)
9898
NNVM_REGISTER_OP(exp)
9999
.describe("take exponential")
100100
.set_num_inputs(1)
101-
.attr("inplace_pair", std::make_pair(0, 0))
102101
.attr<FInferShape>("FInferShape", SameShape);
103102

104103
NNVM_REGISTER_OP(cross_device_copy)

nnvm/tests/python/test_graph.py

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -49,20 +49,6 @@ def test_infer_shape():
4949
assert g.json_attr('shape')[jnode_row_ptr[nindex["reshape1"]]] == [2, 4]
5050
assert g.json_attr('shape')[jnode_row_ptr[nindex["add1"]]] == [4, 2]
5151

52-
def test_infer_shape():
53-
x = sym.Variable('x', shape=(4, 2))
54-
y = sym.add(x, x, name='add1')
55-
y = sym.reshape(y, target=(2, 4), name="reshape1")
56-
g = graph.create(y)
57-
g._set_json_attr("shape_attr_key", "shape")
58-
g = g.apply('InferShape')
59-
jgraph = json.loads(g.apply('SaveJSON').json_attr('json'))
60-
jnodes = jgraph['nodes']
61-
jnode_row_ptr = jgraph['node_row_ptr']
62-
nindex = {n['name']: i for i, n in enumerate(jnodes)}
63-
assert g.json_attr('shape')[jnode_row_ptr[nindex["reshape1"]]] == [2, 4]
64-
assert g.json_attr('shape')[jnode_row_ptr[nindex["add1"]]] == [4, 2]
65-
6652
def test_infer_type():
6753
x = sym.Variable('x')
6854
y = sym.add(x, x, name='add1')

0 commit comments

Comments
 (0)