From fa06aeec62124aa76f4cf7e80172be3e925ef252 Mon Sep 17 00:00:00 2001 From: Jim Lin Date: Fri, 15 Dec 2023 09:37:44 -0800 Subject: [PATCH] #tensorflow fix breakage caused from PR #62362 PiperOrigin-RevId: 591279603 --- site/en/guide/extension_type.ipynb | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/site/en/guide/extension_type.ipynb b/site/en/guide/extension_type.ipynb index 76e20e8d283..7e8edeea7c9 100644 --- a/site/en/guide/extension_type.ipynb +++ b/site/en/guide/extension_type.ipynb @@ -1822,13 +1822,17 @@ " transpose_a=False, transpose_b=False,\n", " adjoint_a=False, adjoint_b=False,\n", " a_is_sparse=False, b_is_sparse=False,\n", - " output_type=None):\n", + " output_type=None,\n", + " grad_a=False, grad_b=False,\n", + " name=None,\n", + " ):\n", " if isinstance(a, MaskedTensor):\n", " a = a.with_default(0)\n", " if isinstance(b, MaskedTensor):\n", " b = b.with_default(0)\n", " return tf.matmul(a, b, transpose_a, transpose_b, adjoint_a,\n", - " adjoint_b, a_is_sparse, b_is_sparse, output_type)" + " adjoint_b, a_is_sparse, b_is_sparse,\n", + " output_type)" ] }, {