Home
last modified time | relevance | path

Searched refs:sp_b (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dsets_test.py184 sp_b = _dense_to_sparse(b_values, dtype=dtype)
185 intersection = self._set_intersection(a, sp_b)
192 self.assertAllEqual(expected_counts, self._set_intersection_count(a, sp_b))
196 intersection = self._set_intersection(sp_a, sp_b)
204 self._set_intersection_count(sp_a, sp_b))
257 sp_b = _dense_to_sparse(b_values, dtype=dtype)
258 intersection = self._set_intersection(a, sp_b)
265 self.assertAllEqual(expected_counts, self._set_intersection_count(a, sp_b))
269 intersection = self._set_intersection(sp_a, sp_b)
277 self._set_intersection_count(sp_a, sp_b))
[all …]
Dsparse_matmul_op_test.py49 sp_b=False, argument
61 b_is_sparse=sp_b)
118 for sp_b in [True, False]:
130 sp_b,
137 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument
156 b_is_sparse=sp_b)
174 for sp_b in [True, False]:
183 name = "sparse_matmul_%s_%s_%s_%s" % (tr_a, tr_b, sp_a, sp_b)
184 self._testGradients(tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype,
Dsparse_concat_op_test.py165 for sp_b in (self._SparseTensorValue_3x5(), self._SparseTensor_3x5()):
167 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b])
214 sp_b = self._SparseTensor_3x5()
218 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b, sp_c])
239 sp_b = self._SparseTensor_String3x5()
242 sp_concat = sparse_ops.sparse_concat(concat_dim, [sp_a, sp_b])
285 sp_b = self._SparseTensor_3x5()
290 [sp_a, sp_b, sp_c, sp_d])
299 sp_b = self._SparseTensor_3x5()
305 concat_dim0, [sp_a, sp_b, sp_c, sp_d], expand_nonconcat_dim=True)
[all …]
Dsparse_add_op_test.py91 for sp_b in (self._SparseTensorValue_3x3(), self._SparseTensor_3x3()):
92 sp_sum = sparse_ops.sparse_add(sp_a, sp_b)
105 sp_b = self._SparseTensor_3x3(negate=True)
107 sp_sum = sparse_ops.sparse_add(sp_a, sp_b, 0.1)
118 sp_b = self._SparseTensor_3x3_v2()
126 sp_sum = sparse_ops.sparse_add(sp_a, sp_b, thresh=0.21)
135 sp_sum = sparse_ops.sparse_add(sp_a, sp_b, thresh=0.11)
150 sp_b, nnz_b = self._randomTensor([n, m], np.float32)
151 sp_sum = sparse_ops.sparse_add(sp_a, sp_b)
155 [sp_a.values, sp_b.values], [(nnz_a,), (nnz_b,)], sp_sum.values,
Dsparse_ops_test.py990 sp_b, unused_b_nnz = _sparsify(b_np, thresh=-.5)
993 maximum_tf = sparse_ops.sparse_maximum(sp_a, sp_b)
996 minimum_tf = sparse_ops.sparse_minimum(sp_a, sp_b)
1001 b_densified = sparse_ops.sparse_tensor_to_dense(sp_b).eval()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py2715 def sparse_maximum(sp_a, sp_b, name=None): argument
2747 [sp_a.indices, sp_a.values, sp_b.indices, sp_b.values]) as name:
2752 sp_b.indices,
2753 sp_b.values,
2754 sp_b.dense_shape,
2761 def sparse_minimum(sp_a, sp_b, name=None): argument
2791 [sp_a.indices, sp_a.values, sp_b.indices, sp_b.values]) as name:
2796 sp_b.indices,
2797 sp_b.values,
2798 sp_b.dense_shape,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt53 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
57 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt57 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
65 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
Dtensorflow.pbtxt2169 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
2177 … argspec: "args=[\'sp_a\', \'sp_b\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "