Fix incorrect display of keras model summary

This fix fixes the issue raised in 24627 where the
display of keras model summary is incorrect (regression
from 1.12). The reason was that `layer.name`
(vs. layer object itself) should be used,

This fix fixes 24627.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>