saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-04 04:07:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add gi overload (#9690)

Browse source
This commit is contained in:
ashari4 2021-11-07 16:04:00 -08:00 committed by GitHub
parent c6fddb263f
commit 1151c661eb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
orttraining/orttraining/core/graph/gradient_builder_base.h
View file

@ -139,6 +139,12 @@ class GradientBuilderBase {
return ArgDef(GradientName(node_->InputDefs()[i]->Name()), node_->InputDefs()[i]->TypeAsProto());
}
// gradient of i-th input of forward op - useful when gradient type does not match input type
ArgDef GI(const size_t i, const TypeProto *type) const {
ORT_ENFORCE(i < node_->InputDefs().size());
return ArgDef(GradientName(node_->InputDefs()[i]->Name()), type);
}
// gradient of i-th output of forward op
ArgDef GO(const size_t i) const {
ORT_ENFORCE(i < node_->OutputDefs().size());