onnxruntime
Added RequiredGrad attribute to YieldOp
#6657
Merged

Added RequiredGrad attribute to YieldOp #6657

satyajandhyala
satyajandhyala satyajandhyala force pushed from cc28f5b6 to 51380f78 5 years ago
satyajandhyala satyajandhyala force pushed from 51380f78 5 years ago
satyajandhyala satyajandhyala force pushed to 5f5dbed2 5 years ago
satyajandhyala satyajandhyala force pushed from 5f5dbed2 to fa5b2b80 5 years ago
satyajandhyala satyajandhyala force pushed 5 years ago
satyajandhyala satyajandhyala force pushed 5 years ago
satyajandhyala satyajandhyala force pushed to 8002efbe 5 years ago
satyajandhyala satyajandhyala marked this pull request as ready for review 5 years ago
satyajandhyala satyajandhyala requested a review from baijumeswani baijumeswani 5 years ago
satyajandhyala satyajandhyala requested a review from BowenBao BowenBao 5 years ago
satyajandhyala satyajandhyala requested a review from liqunfu liqunfu 5 years ago
satyajandhyala satyajandhyala requested a review from spandantiwari spandantiwari 5 years ago
satyajandhyala satyajandhyala requested a review from thiagocrepaldi thiagocrepaldi 5 years ago
satyajandhyala satyajandhyala requested a review 5 years ago
satyajandhyala satyajandhyala assigned satyajandhyala satyajandhyala 5 years ago
satyajandhyala satyajandhyala changed the title [WIP] Added RequiredGrad attribute to YieldOp Added RequiredGrad attribute to YieldOp 5 years ago
SherlockNoMad SherlockNoMad added component:ortmodule
satyajandhyala satyajandhyala force pushed to d8cebe2d 5 years ago
satyajandhyala satyajandhyala force pushed from 217f85e5 to f346f480 5 years ago
Lafi7e
Lafi7e commented on 2021-02-24
Lafi7e
Lafi7e approved these changes on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
SherlockNoMad
SherlockNoMad commented on 2021-02-24
satyajandhyala satyajandhyala force pushed from c4f9ccea to e6345ac2 5 years ago
satyajandhyala satyajandhyala force pushed from e6345ac2 to 0f8afee2 5 years ago
satyajandhyala satyajandhyala force pushed from 0f8afee2 to 2d28f45b 5 years ago
satyajandhyala satyajandhyala force pushed from 69841faa to ef733edd 5 years ago
satyajandhyala satyajandhyala force pushed from ef733edd to d7a0ac67 5 years ago
satyajandhyala satyajandhyala force pushed from d7a0ac67 to 44855cf0 5 years ago
satyajandhyala satyajandhyala force pushed from f3f492cb to ed576062 5 years ago
SherlockNoMad
SherlockNoMad commented on 2021-02-25
satyajandhyala Added RequiredGrad attribute to YieldOp
6d95b796
satyajandhyala Added more testcases.
e913f37e
satyajandhyala Check node_arg is valid before adding to input_args.
1206a712
satyajandhyala Added constraint on the number of YieldOp outputs.
dc54bef4
satyajandhyala Chagened YieldOp attribute to hold the indices of the required gradie…
f3a048c3
satyajandhyala Changed backward_output_grad_names to a map from backward output grad…
ce86d3d1
satyajandhyala Removed code saving ONNX file.
607593a0
satyajandhyala Added conversion from size_t to int.
4df00258
satyajandhyala Fixed the RequiredGrad attribute description.
e25778cd
satyajandhyala Changed not_arg type
a1fd3823
satyajandhyala Fixed input index and added/fixed assertions.
c7f45835
satyajandhyala Removed unnecessary variables.
6382b0ad
satyajandhyala Removed changes to obsolete file.
20618edd
satyajandhyala Change attribute name RequiredGrad to required_grad.
ba0d7c27
satyajandhyala Rolled multiple test cases into one and targetting only cuda, not cpu.
0e18d1fd
satyajandhyala Added comment
33395c13
satyajandhyala Changed map to unordered_map
80626e99
satyajandhyala Changed non-null condition into assertion.
f0e4e05c
satyajandhyala Changed the order and numbers of the tests.
e9f916fe
satyajandhyala Update module_gradient_graph_builder.h
f4297935
satyajandhyala satyajandhyala force pushed from ed576062 to f4297935 5 years ago
satyajandhyala Added more assertions to check the gradients and compare gradients ag…
89d96abc
satyajandhyala satyajandhyala force pushed from 8ffa297b to 89d96abc 5 years ago
SherlockNoMad
SherlockNoMad commented on 2021-02-26
SherlockNoMad
SherlockNoMad approved these changes on 2021-02-26
satyajandhyala satyajandhyala merged 355057cf into thiagofc/ortmodule-api 5 years ago
satyajandhyala satyajandhyala deleted the ortmodule_api/yield_op_multiple_input_output branch 5 years ago

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone