-
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
[PaddleSpeech] Add OPs and others needed by fastspeech_2 model #9706
[PaddleSpeech] Add OPs and others needed by fastspeech_2 model #9706
Conversation
test=develop
…educe_max OP kInt32 registration test=develop
test=develop
Thanks for your contribution! |
test=develop
9a0613d
to
094f36f
Compare
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
if (step > 0) { | ||
slice_dims[axis] = (end - start + step - 1) / step; | ||
} else { | ||
slice_dims[axis] = (end - start + step + 1) / step; |
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.
这个step逻辑是不是有问题,step = 0可以避免吗?
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.
这里的处理逻辑是与paddle保持一致。
…ePaddle#9706) * [Host] add 3 OPs: set_value, round, share_data test=develop * [Host] add expand_v2 OP registration with type kBool test=develop * [Arm] add reduce_sum OP Int64 registration and neon implement & add reduce_max OP kInt32 registration test=develop * [X86] fix bug in set_value OP test=develop * [Extra] move 2 round and share_data to extra test=develop
…ePaddle#9706) * [Host] add 3 OPs: set_value, round, share_data test=develop * [Host] add expand_v2 OP registration with type kBool test=develop * [Arm] add reduce_sum OP Int64 registration and neon implement & add reduce_max OP kInt32 registration test=develop * [X86] fix bug in set_value OP test=develop * [Extra] move 2 round and share_data to extra test=develop
* [X86] Add set value op and double data type to framework. (#9580) * [QualcommQnn] add ops (#9538) support fusion_elementwise_mul_activation, fusion_elementwise_sub_activation, fusion_elementwise_div_activation, fusion_elementwise_min_activation, fusion_elementwise_max_activation, fusion_elementwise_pow_activation, instance_norm, prelu, arg_max, arg_min, flatten, flatten2, norm * add float64 type to lite * add float64 kernel for set value * change the third-party-libs url due to flatbuf update. * fix include files conflict * fix bug * Fix heterogeneous execution errors * fix control_flow_op_control_flow_op_shared_inputs_and_outputs_place_sync_pass bug * fix comment Co-authored-by: zhupengyang <[email protected]> * [PaddleSpeech] Add OPs and others needed by fastspeech_2 model (#9706) * [Host] add 3 OPs: set_value, round, share_data test=develop * [Host] add expand_v2 OP registration with type kBool test=develop * [Arm] add reduce_sum OP Int64 registration and neon implement & add reduce_max OP kInt32 registration test=develop * [X86] fix bug in set_value OP test=develop * [Extra] move 2 round and share_data to extra test=develop * [proto] fix a bug test=develop Co-authored-by: csy0225 <[email protected]> Co-authored-by: zhupengyang <[email protected]>
[Host] Add 3 OPs: set_value, share_data, round;
[Host] Add expand_v2 Bool registration .
[ARM] Add reduce_max OP INT32 registration;
[ARM] Add reduce_sum OP INT64 registration and neon implement.
[X86] Fix a bug in set_value OP.