Closed MasterJH5574 closed 1 year ago
As tracked by #332, this PR is the O2f milestone of the high-level operator introduction plan.
For the first part, this PR rewrites our current set operator: unique. Changes including
unique
return_index
dim
axis
tensor.py
set.py
op_attr_types.h
set.h
test_op_set.py
For the second part, this PR introduces one searching operator - where.
where
The two operators are well-tested.
As tracked by #332, this PR is the O2f milestone of the high-level operator introduction plan.
For the first part, this PR rewrites our current set operator:
unique. Changes includingreturn_indexwhich exists in NumPy and ONNX’s API; renamingdimtoaxisto align with NumPy.tensor.pyto standaloneset.py.op_attr_types.hto standaloneset.h.test_op_set.py.For the second part, this PR introduces one searching operator -
where.The two operators are well-tested.