Merge pull request #12439 from jacquesqiao/CheckTensorNANOrInf-support-selectedrows

CheckTensorNANOrInf support checking SelectedRows
bugfix/anakin-compile
Qiao Longfei 7 years ago committed by GitHub
commit 297cbeb1c6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -679,6 +679,8 @@ void OperatorWithKernel::RunImpl(const Scope& scope,
if (var == nullptr) continue;
if (var->IsType<framework::LoDTensor>()) {
CheckTensorNANOrInf(vname, var->Get<framework::LoDTensor>());
} else if (var->IsType<framework::SelectedRows>()) {
CheckTensorNANOrInf(vname, var->Get<framework::SelectedRows>().value());
}
}
}

Loading…
Cancel
Save