I try to make a custom layer. Everything work ok in Eager mode and successfully compiled. But when I try to fit my model I\'ve got the next error: I recognized that the reason i