-
Notifications
You must be signed in to change notification settings - Fork 1.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add unique_op #9819
add unique_op #9819
Conversation
Thanks for your contribution! |
lite/operators/unique_op.cc
Outdated
} | ||
|
||
bool UniqueOp::InferShapeImpl() const { | ||
DDim in_dims = param_.X->dims(); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
// automatically generated by the FlatBuffers compiler, do not modify | ||
|
||
|
||
#ifndef FLATBUFFERS_GENERATED_TUNECACHE_PADDLE_LITE_FBS_OPENCL_PROTO_H_ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个文件也不要提交
lite/core/program.cc
Outdated
@@ -815,6 +797,231 @@ void Instruction::Run() { | |||
kernel_->Launch(); | |||
has_run_ = true; | |||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个代码就不用交了
st.integers( | ||
min_value=2, max_value=100), | ||
min_size=1, | ||
max_size=1)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
input_shape 多测几个维度,下面的print去掉
pass | ||
|
||
def test(self, *args, **kwargs): | ||
self.run_and_statis(quant=False, max_examples=25) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
提交代码可以写25,自测的时候要多测几组数据
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
PR Type
Add new OP
PR Change
operator && kernel
PR Describe
Add unique OP on Host backends and fix index_select OP bug