Skip to content
Snippets Groups Projects
Commit be217da2 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Merge branch 'bindLoss' into 'dev'

Update how loss function work

See merge request !60
parents 0632fa8c d5b6a145
No related branches found
No related tags found
2 merge requests!610.2.2,!60Update how loss function work
Pipeline #45154 failed
...@@ -72,7 +72,10 @@ void Aidge::FCImpl_cpu::backward() ...@@ -72,7 +72,10 @@ void Aidge::FCImpl_cpu::backward()
{ {
const FC_Op& op_ = dynamic_cast<const FC_Op&>(mOp); const FC_Op& op_ = dynamic_cast<const FC_Op&>(mOp);
const auto& fc_grad = op_.getOutput(0)->grad(); const auto& fc_grad = op_.getOutput(0)->grad();
assert(fc_grad && "missing ouput #0 gradient"); AIDGE_ASSERT(fc_grad, "missing ouput #0 gradient");
AIDGE_ASSERT(op_.getInput(0)->grad(), "missing input #0 gradient");
AIDGE_ASSERT(op_.getInput(1)->grad(), "missing input #1 gradient");
AIDGE_ASSERT(op_.getInput(2)->grad(), "missing input #2 gradient");
// Find the correct kernel type // Find the correct kernel type
const Registrar<FCImplBackward_cpu>::registrar_key registrarKey = { const Registrar<FCImplBackward_cpu>::registrar_key registrarKey = {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment