Add num_summed_dimensions to get_config

f3a6500f · Yichao 'Peak' Ji · GitHub · f41682df · f3a6500f
Unverified Commit f3a6500f authored Apr 08, 2020 by Yichao 'Peak' Ji Committed by GitHub Apr 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/nlp/modeling/layers/dense_einsum.py official/nlp/modeling/layers/dense_einsum.py +2 -0

No files found.
--- a/official/nlp/modeling/layers/dense_einsum.py
+++ b/official/nlp/modeling/layers/dense_einsum.py
@@ -147,6 +147,8 @@ class DenseEinsum(tf.keras.layers.Layer):
    config = {
        "output_shape":
            self._output_shape,
+        "num_summed_dimensions":
+            self._num_summed_dimensions,
        "activation":
            tf.keras.activations.serialize(self._activation),
        "use_bias":