Skip to content
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

Dev optimize tensor setitem #5501

Merged
merged 19 commits into from
Jul 16, 2021
Merged

Conversation

hjchen2
Copy link
Contributor

@hjchen2 hjchen2 commented Jul 15, 2021

No description provided.

return flow.F.expand(
x, in_shape=list(x.shape), out_shape=new_size, stride=new_stride
)
return flow.F.expand(x, self.expand_size)
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这里重新实现了一下expand接口

oneflow/core/functional/tensor_index.cpp Show resolved Hide resolved
oneflow/python/framework/tensor.py Show resolved Hide resolved
@hjchen2 hjchen2 requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 15, 2021 07:07
@hjchen2 hjchen2 requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 15, 2021 07:19
@hjchen2 hjchen2 force-pushed the dev_optimize_tensor_setitem branch from 5207f86 to 6fb1875 Compare July 15, 2021 14:09
@hjchen2 hjchen2 requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 15, 2021 14:10
@oneflow-ci-bot oneflow-ci-bot removed their request for review July 16, 2021 03:44
@oneflow-ci-bot oneflow-ci-bot self-requested a review July 16, 2021 09:46
@oneflow-ci-bot oneflow-ci-bot requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 16, 2021 10:45
@oneflow-ci-bot oneflow-ci-bot requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 16, 2021 12:13
@oneflow-ci-bot oneflow-ci-bot requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 16, 2021 13:38
@oneflow-ci-bot oneflow-ci-bot requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 16, 2021 15:23
@oneflow-ci-bot oneflow-ci-bot self-requested a review July 16, 2021 17:15
@oneflow-ci-bot oneflow-ci-bot requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 16, 2021 18:43
@oneflow-ci-bot oneflow-ci-bot requested review from oneflow-ci-bot and removed request for oneflow-ci-bot July 16, 2021 21:40
@oneflow-ci-bot oneflow-ci-bot merged commit 61a951e into master Jul 16, 2021
@oneflow-ci-bot oneflow-ci-bot deleted the dev_optimize_tensor_setitem branch July 16, 2021 22:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants