try to move shape_array to npx (#16897)

apache · Nov 25, 2019 · 9b49cfe · 9b49cfe
1 parent 11edb70
commit 9b49cfe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/operator/tensor/elemwise_unary_op_basic.cc b/src/operator/tensor/elemwise_unary_op_basic.cc
@@ -565,6 +565,7 @@ void ShapeComputeCPU(const nnvm::NodeAttrs& attrs,
 }
 
 NNVM_REGISTER_OP(shape_array)
+.add_alias("_npx_shape_array")
 .describe(R"code(Returns a 1D int64 array containing the shape of data.
 
 Example::