Skip to content

GitLab

Commit 5b095712 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Merge pull request #8372 from peakji:master 

PiperOrigin-RevId: 305725472
parents 2f5e8e3d f3a6500f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/modeling/layers/dense_einsum.py
View file @ 5b095712
...@@ -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