diff --git a/tensorflow/python/ops/math_ops.py b/tensorflow/python/ops/math_ops.py index a0d3f06627c..c8de4a5dbc5 100644 --- a/tensorflow/python/ops/math_ops.py +++ b/tensorflow/python/ops/math_ops.py @@ -2258,7 +2258,7 @@ def matvec(a, _OverrideBinaryOperatorHelper(matmul, "matmul") sparse_matmul = gen_math_ops.sparse_mat_mul -tf_export("sparse_matmul")(sparse_matmul) +tf_export(v1=["sparse_matmul"])(sparse_matmul) @ops.RegisterStatistics("MatMul", "flops") diff --git a/tensorflow/tools/api/golden/v2/tensorflow.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.pbtxt index 5293262e7f8..91d0a4efcec 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.pbtxt @@ -1140,10 +1140,6 @@ tf_module { name: "sparse_concat" argspec: "args=[\'axis\', \'sp_inputs\', \'name\', \'expand_nonconcat_dim\', \'concat_dim\'], varargs=None, keywords=None, defaults=[\'None\', \'False\', \'None\'], " } - member_method { - name: "sparse_matmul" - argspec: "args=[\'a\', \'b\', \'transpose_a\', \'transpose_b\', \'a_is_sparse\', \'b_is_sparse\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'False\', \'False\', \'False\', \'None\'], " - } member_method { name: "sparse_reduce_max" argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "