experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Only export tf.sparse_matmul in V1. This is replaced by tf.linalg.matmul in V2.

Browse Source 
PiperOrigin-RevId: 221103660
This commit is contained in:
A. Unique TensorFlower 2018-11-12 09:21:42 -08:00 committed by TensorFlower Gardener
parent c7c62b6f74
commit b62da0d3fa
2 changed files with 1 additions and 5 deletions
tensorflow
python/ops
math_ops.py
tools/api/golden/v2
tensorflow.pbtxt

2
tensorflow/python/ops/math_ops.py
View File

@ -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")

4
tensorflow/tools/api/golden/v2/tensorflow.pbtxt
View File

@ -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\'], "