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

Commit

Permalink
Param is moved
Browse files Browse the repository at this point in the history
  • Loading branch information
mozga-intel committed Sep 3, 2021
1 parent 6dce172 commit 8f0bea4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/api/operator/numpy/random/np_choice_op.cc
Original file line number Diff line number Diff line change
Expand Up @@ -33,11 +33,11 @@ MXNET_REGISTER_API("_npi.choice")
using namespace runtime;
const nnvm::Op* op = Op::Get("_npi_choice");
nnvm::NodeAttrs attrs;
op::NumpyChoiceParam param = nnvm::get<op::NumpyChoiceParam>(attrs.parsed);

NDArray* inputs[2];
int num_inputs = 0;

op::NumpyChoiceParam param;
if (args[0].type_code() == kDLInt) {
param.a = args[0].operator int();
} else if (args[0].type_code() == kNDArrayHandle) {
Expand Down

0 comments on commit 8f0bea4

Please sign in to comment.