diff --git a/oneflow/python/ops/array_ops.py b/oneflow/python/ops/array_ops.py
index b25faeb61802eee82e283caa1e5678e1547ddd63..918b0e5564326688899e85b088839d5048dc4048 100644
--- a/oneflow/python/ops/array_ops.py
+++ b/oneflow/python/ops/array_ops.py
@@ -363,7 +363,7 @@ def piece_slice(inputs, output_size, name=None):
 
 
 @oneflow_export("elem_cnt")
-def elem_cnt(inputs, begin_axis=None, end_axis=None, dtype=None, name=None):
+def elem_cnt(inputs, dtype=None, name=None):
     op_conf = op_conf_util.OperatorConf()
     setattr(op_conf, "name", name if name is not None else id_util.UniqueStr("ElemCnt_"))
     op_conf.shape_elem_cnt_conf.x = inputs.logical_blob_name
@@ -417,7 +417,7 @@ def stack(inputs, axis, name=None):
 
 
 @oneflow_export("assign")
-def assign(ref, value, begin_axis=None, end_axis=None, dtype=None, name=None):
+def assign(ref, value, dtype=None, name=None):
     op_conf = op_conf_util.OperatorConf()
     setattr(op_conf, "name", name if name is not None else id_util.UniqueStr("Assign_"))
     op_conf.assign_conf.ref = ref.logical_blob_name