Add registry function and tests for `tf.keras.layers.EinsumDense`.#431

copybara-service[bot] wants to merge 1 commit intomastertensorflow/privacy:masterfrom

cl/511864469tensorflow/privacy:cl/511864469Copy head branch name to clipboard