Searched refs:sp_t (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 631 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
632 densified = self.evaluate(sparse_ops.sparse_tensor_to_dense(sp_t))
656 tf_dense_ans = sparse_ops.sparse_reduce_sum(sp_t, reduction_axes,
659 tf_dense_ans = sparse_ops.sparse_reduce_max(sp_t, reduction_axes,
664 tf_sparse_ans = sparse_ops.sparse_reduce_sum_sparse(sp_t,
668 tf_sparse_ans = sparse_ops.sparse_reduce_max_sparse(sp_t,
677 def _compare_all(self, sp_t, reduction_axes, ndims): argument
678 self._compare(sp_t, reduction_axes, ndims, False, False)
679 self._compare(sp_t, reduction_axes, ndims, False, True)
680 self._compare(sp_t, reduction_axes, ndims, True, False)
[all …]
|
| D | sparse_tensor_dense_matmul_grad_test.py | 65 sp_t, nnz = self._randomTensor(
74 sp_t, dense_t, adjoint_a=adjoint_a, adjoint_b=adjoint_b, name=name)
80 [dense_t, sp_t.values], [dense_t_shape, sp_t_val_shape], matmul,
|
| D | sparse_add_op_test.py | 222 sp_t, unused_nnz = _sparsify(sp_vals, thresh=sparsity, index_dtype=np.int32)
226 sparse_ops.sparse_tensor_to_dense(sp_t), constant_op.constant(vals))
227 sa = sparse_ops.sparse_add(sp_t, constant_op.constant(vals))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 792 def sparse_dense_cwise_add(sp_t, dense_t): argument
812 result = gen_sparse_ops.sparse_dense_cwise_add(sp_t.indices, sp_t.values,
813 sp_t.dense_shape, dense_t)
814 return sparse_tensor.SparseTensor(sp_t.indices, result, sp_t.dense_shape)
|