Added RequiredGrad attribute to YieldOp #6657
satyajandhyala
marked this pull request as ready for review 5 years ago
satyajandhyala
changed the title [WIP] Added RequiredGrad attribute to YieldOp Added RequiredGrad attribute to YieldOp 5 years ago
Lafi7e
commented
on 2021-02-24
Lafi7e
approved these changes
on 2021-02-24
Added RequiredGrad attribute to YieldOp
6d95b796
Added more testcases.
e913f37e
Check node_arg is valid before adding to input_args.
1206a712
Added constraint on the number of YieldOp outputs.
dc54bef4
Chagened YieldOp attribute to hold the indices of the required gradie…
f3a048c3
Changed backward_output_grad_names to a map from backward output grad…
ce86d3d1
Removed code saving ONNX file.
607593a0
Added conversion from size_t to int.
4df00258
Fixed the RequiredGrad attribute description.
e25778cd
Changed not_arg type
a1fd3823
Fixed input index and added/fixed assertions.
c7f45835
Removed unnecessary variables.
6382b0ad
Removed changes to obsolete file.
20618edd
Change attribute name RequiredGrad to required_grad.
ba0d7c27
Rolled multiple test cases into one and targetting only cuda, not cpu.
0e18d1fd
Added comment
33395c13
Changed map to unordered_map
80626e99
Changed non-null condition into assertion.
f0e4e05c
Changed the order and numbers of the tests.
e9f916fe
Update module_gradient_graph_builder.h
f4297935
Added more assertions to check the gradients and compare gradients ag…
89d96abc
satyajandhyala
merged
355057cf
into thiagofc/ortmodule-api 5 years ago
satyajandhyala
deleted the ortmodule_api/yield_op_multiple_input_output branch 5 years ago
Login to write a write a comment.
Login via GitHub