Internal change

PiperOrigin-RevId: 530426973

Internal change
PiperOrigin-RevId: 530426973
ff2069de · James Lee-Thorp · A. Unique TensorFlower · 954dd73b · ff2069de
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

official/nlp/modeling/layers/mixing.py official/nlp/modeling/layers/mixing.py +2 -2

未找到文件。
--- a/official/nlp/modeling/layers/mixing.py
+++ b/official/nlp/modeling/layers/mixing.py
@@ -281,5 +281,5 @@ def _pick_fourier_transform(

    return functools.partial(
        two_dim_matmul,
-        matrix_dim_one=tf.convert_to_tensor(dft_mat_seq),
-        matrix_dim_two=tf.convert_to_tensor(dft_mat_hidden))
+        matrix_dim_one=dft_mat_seq,
+        matrix_dim_two=dft_mat_hidden)