Week 3 assignment with tensorflow 3.1 - Implement Forward Propagation

in the section “3.1 - Implement Forward Propagation”
Exercise 5 - forward_propagation where we implement forward_propagation function

I implemented it like this
image

But when asserted, it says “Output does not match”
I tried running all cells above, all got correct answers and now I still cannot spot what’s wrong here…

Can anyone help, please?

Not sure what’s been wrong but when I submitted the assignment, I got 100/100.

I imagine that the values used for the TF tensor in forward_propagation_test() (which tests forward_propagation() just one cell below) are incorrect ones.