-
Notifications
You must be signed in to change notification settings - Fork 157
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
[onert] Plan gradient tensors on train backend #13523
[onert] Plan gradient tensors on train backend #13523
Conversation
This commit adds planning gradient tensors used on train backend. ONE-DCO-1.0-Signed-off-by: ragmani <[email protected]>
a2c4379
to
9447a61
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
// TODO Plan gradient tensors | ||
VERBOSE(BackendContext) << "Start planning gradient tensors" << std::endl; | ||
|
||
// TODO Use DisposableTensor instread of GradientTensor to plan them together if possible |
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.
// TODO Use DisposableTensor instread of GradientTensor to plan them together if possible | |
// TODO Use DisposableTensor instead of GradientTensor to plan them together if possible |
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.
@ragmani Please fix typo later
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 👍
This commit adds planning gradient tensors used on train backend.
ONE-DCO-1.0-Signed-off-by: ragmani [email protected]