Skip to content

GitLab

Unverified Commit f3a6500f authored by Yichao 'Peak' Ji's avatar Yichao 'Peak' Ji Committed by GitHub
Browse files

Add num_summed_dimensions to get_config

parent f41682df
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/modeling/layers/dense_einsum.py
View file @ f3a6500f
...@@ -147,6 +147,8 @@ class DenseEinsum(tf.keras.layers.Layer): ...@@ -147,6 +147,8 @@ class DenseEinsum(tf.keras.layers.Layer):
config = { config = {
"output_shape": "output_shape":
self._output_shape, self._output_shape,
"num_summed_dimensions":
self._num_summed_dimensions,
"activation": "activation":
tf.keras.activations.serialize(self._activation), tf.keras.activations.serialize(self._activation),
"use_bias": "use_bias":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment