[custom-layer] Update API usage for serialization (#273)
[custom-layer] Update API usage for serialization (#273)

* [custom-layer] Update API usage for serialization

tensorflow/tfjs-examples,tensorflow/tfjs-examples,tensorflow/tfjs-examples,tensorflow/tfjs-examples,tensorflow/tfjs-examples,tensorflow/tfjs-examples,tensorflow/tfjs-examples