Skip to content

Commit

Permalink
change api attribute name: pool_type to reduce_op, compute_type to me…
Browse files Browse the repository at this point in the history
…ssage_op
  • Loading branch information
DesmonDay committed Aug 11, 2022
1 parent 50bf7da commit e7cbc9f
Show file tree
Hide file tree
Showing 3 changed files with 94 additions and 97 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -25,11 +25,11 @@
def graph_send_recv_wrapper(x,
src_index,
dst_index,
pool_type="sum",
reduce_op="sum",
out_size=None,
name=None):
return paddle.geometric.send_u_recv(x, src_index, dst_index,
pool_type.lower(), out_size, name)
reduce_op.lower(), out_size, name)


class TestGraphSendRecvMaxOp(OpTest):
Expand Down
Loading

0 comments on commit e7cbc9f

Please sign in to comment.