What's the recommended way to have this layout respect autolayout?

Could you please explain how it would be possible to return the correct size driven by autolayout on my collection view cells? 
I have a layout which scrolls vertically with a fixed number of columns. So the correct height would need to be returned in order to render my views correctly.

Any help on this would be greatly appreciated. 
Thank you