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

Commit

Permalink
fix code style
Browse files Browse the repository at this point in the history
  • Loading branch information
gyshi committed Sep 5, 2019
1 parent 799d916 commit 71e7e15
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
12 changes: 6 additions & 6 deletions src/operator/numpy/np_matrix_op.cc
Expand Up @@ -422,12 +422,12 @@ Other axes remain in their original order.
.set_attr<nnvm::FGradient>("FGradient",
[](const nnvm::NodePtr& n, const std::vector<nnvm::NodeEntry>& ograds) {
const NumpyMoveaxisParam& param = nnvm::get<NumpyMoveaxisParam>(n->attrs.parsed);
std::ostringstream os1;
os1 << param.source;
std::ostringstream os2;
os2 << param.destination;
return MakeNonlossGradNode("_np_moveaxis", n, ograds, {}, \
{{"source", os2.str()}, {"destination", os1.str()}});
std::ostringstream os1;
os1 << param.source;
std::ostringstream os2;
os2 << param.destination;
return MakeNonlossGradNode("_np_moveaxis", n, ograds, {},
{{"source", os2.str()}, {"destination", os1.str()}});
})
.set_attr<FCompute>("FCompute<cpu>", NumpyMoveaxisCompute<cpu>)
.set_attr<nnvm::FListInputNames>("FListInputNames",
Expand Down
2 changes: 1 addition & 1 deletion tests/python/unittest/test_numpy_op.py
Expand Up @@ -1659,7 +1659,7 @@ def __init__(self, source=None, destination=None):
def hybrid_forward(self, F, x):
return F.np.moveaxis(x, source=self._source, destination=self._destination)

dtypes = ['float32', 'int32', 'float64', 'float16', 'int64']
dtypes = ['int32', 'int64', 'float16', 'float32', 'float64']
for hybridize in [False, True]:
for dtype in dtypes:
for ndim in [0, 1, 2, 3, 4, 5, 6]:
Expand Down

0 comments on commit 71e7e15

Please sign in to comment.