Home
last modified time | relevance | path

Searched refs:sp_a (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dsets_test.py191 sp_a = _dense_to_sparse(a_values, dtype=dtype)
192 intersection = self._set_intersection(sp_a, sp_b)
200 self._set_intersection_count(sp_a, sp_b))
262 sp_a = _dense_to_sparse(a_values, dtype=dtype)
263 intersection = self._set_intersection(sp_a, sp_b)
271 self._set_intersection_count(sp_a, sp_b))
314 sp_a = sparse_tensor_lib.SparseTensor(
365 self._set_intersection(sp_a, sp_b)
410 intersection = self._set_intersection(sp_a, sp_b)
418 self._set_intersection_count(sp_a, sp_b))
[all …]
Dsparse_concat_op_test.py140 for sp_a in (self._SparseTensorValue_3x3(), self._SparseTensor_3x3()):
144 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a])
163 for sp_a in (self._SparseTensorValue_3x3(), self._SparseTensor_3x3()):
166 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b])
188 sp_a = self._SparseTensor_3x3()
192 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_d])
212 sp_a = self._SparseTensor_3x3()
217 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b, sp_c])
237 sp_a = self._SparseTensor_String3x3()
241 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b])
[all …]
Dsparse_matmul_op_test.py47 sp_a=True, argument
59 a_is_sparse=sp_a,
111 for sp_a in [True, False]:
123 sp_a,
131 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument
149 a_is_sparse=sp_a,
165 for sp_a in [True, False]:
175 name = "sparse_matmul_%s_%s_%s_%s" % (tr_a, tr_b, sp_a, sp_b)
176 self._testGradients(tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype,
Dsparse_add_op_test.py89 for sp_a in (self._SparseTensorValue_3x3(), self._SparseTensor_3x3()):
91 sp_sum = sparse_ops.sparse_add(sp_a, sp_b)
103 sp_a = self._SparseTensor_3x3()
106 sp_sum = sparse_ops.sparse_add(sp_a, sp_b, 0.1)
116 sp_a = self._SparseTensor_3x3()
125 sp_sum = sparse_ops.sparse_add(sp_a, sp_b, thresh=0.21)
134 sp_sum = sparse_ops.sparse_add(sp_a, sp_b, thresh=0.11)
147 sp_a, nnz_a = self._randomTensor([n, m], np.float32)
149 sp_sum = sparse_ops.sparse_add(sp_a, sp_b)
153 [sp_a.values, sp_b.values], [(nnz_a,), (nnz_b,)], sp_sum.values,
Dsparse_ops_test.py891 sp_a, unused_a_nnz = _sparsify(a_np, thresh=-.5)
895 maximum_tf = sparse_ops.sparse_maximum(sp_a, sp_b)
898 minimum_tf = sparse_ops.sparse_minimum(sp_a, sp_b)
902 a_densified = sparse_ops.sparse_tensor_to_dense(sp_a).eval()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py1624 def sparse_tensor_dense_matmul(sp_a, argument
1827 sp_a = _convert_to_sparse_tensor(sp_a)
1829 [sp_a.indices, sp_a.values, b]) as name:
1832 a_indices=sp_a.indices,
1833 a_values=sp_a.values,
1834 a_shape=sp_a.dense_shape,
1896 def sparse_maximum(sp_a, sp_b, name=None): argument
1920 [sp_a.indices, sp_a.values, sp_b.indices, sp_b.values]) as name:
1922 sp_a.indices,
1923 sp_a.values,
[all …]
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.pbtxt1813 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
1821 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
1885 …argspec: "args=[\'sp_a\', \'b\', \'adjoint_a\', \'adjoint_b\', \'name\'], varargs=None, keywords=N…