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

Commit 67acbe5

Browse files
eric-haibin-linTaoLv
authored andcommitted
[DOC] Fix many arguments in the doc: reshape_like, arange_like, shape_array (#15752)
* Update init_op.cc * Update elemwise_unary_op_basic.cc
1 parent 3112893 commit 67acbe5

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/operator/tensor/elemwise_unary_op_basic.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -546,7 +546,8 @@ Negative indices are supported, and `None` can be used for either `lhs_end` or `
546546
return lhs;
547547
})
548548
.add_argument("lhs", "NDArray-or-Symbol", "First input.")
549-
.add_argument("rhs", "NDArray-or-Symbol", "Second input.");
549+
.add_argument("rhs", "NDArray-or-Symbol", "Second input.")
550+
.add_arguments(ReshapeLikeParam::__FIELDS__());
550551

551552
void ShapeComputeCPU(const nnvm::NodeAttrs& attrs,
552553
const OpContext& ctx,
@@ -594,8 +595,7 @@ Example::
594595
TYPE_ASSIGN_CHECK(*out_attrs, 0, mshadow::kInt64);
595596
return out_attrs->at(0) != -1;
596597
})
597-
.add_argument("data", "NDArray-or-Symbol", "Input Array.")
598-
.add_arguments(ReshapeLikeParam::__FIELDS__());
598+
.add_argument("data", "NDArray-or-Symbol", "Input Array.");
599599

600600
void SizeComputeCPU(const nnvm::NodeAttrs& attrs,
601601
const OpContext& ctx,

src/operator/tensor/init_op.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,8 @@ Examples::
134134
[](const NodeAttrs& attrs) { return std::vector<uint32_t>(1, 0); })
135135
.set_attr<FCompute>("FCompute<cpu>", RangeCompute<cpu, RangeLikeParam>)
136136
.set_attr<nnvm::FGradient>("FGradient", MakeZeroGradNodes)
137-
.add_argument("data", "NDArray-or-Symbol", "The input");
137+
.add_argument("data", "NDArray-or-Symbol", "The input")
138+
.add_arguments(RangeLikeParam::__FIELDS__());
138139

139140
NNVM_REGISTER_OP(_linspace)
140141
.describe("Return evenly spaced numbers over a specified interval. Similar to Numpy")

0 commit comments

Comments
 (0)