| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensor_dense_matmul_op_test.py | 147 dense_t = np.matrix([[1] * 5, [2] * 5], dtype=np.float32)
149 self.evaluate(sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
150 dense_t = np.matrix([[1] * 500, [2] * 500], dtype=np.float32)
152 self.evaluate(sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
155 dense_t = np.matrix([[1] * 5, [2] * 5, [3] * 5], dtype=np.float32)
159 sparse_t, dense_t, adjoint_a=True))
160 dense_t = np.matrix([[1] * 500, [2] * 500, [3] * 500], dtype=np.float32)
164 sparse_t, dense_t, adjoint_a=True))
175 dense_t = np.matrix([[1] * 5, [2] * 5], dtype=np.float32)
178 expected_t, sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
[all …]
|
| D | sparse_tensor_dense_matmul_grad_test.py | 71 dense_t = self._randomTensor([k, m], values_dtype, adjoint=adjoint_b)
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_ops_test.py | 829 dense_t = constant_op.constant(dense_vals_np)
831 self._check(sp_t / dense_t, sp_t_densified / dense_vals_np, sp_t)
833 self._check(sp_t * dense_t, sp_t_densified * dense_vals_np, sp_t)
834 self._check(dense_t * sp_t, sp_t_densified * dense_vals_np, sp_t)
837 res = sp_t / dense_t # should invoke "__truediv__"
848 dense_t = array_ops.ones(shape, dtype=dtypes.int32)
850 sparse_ops.sparse_dense_cwise_add(sp_t, dense_t),
854 dense_t = array_ops.ones([1], dtype=dtypes.int32)
856 sparse_ops.sparse_dense_cwise_add(sp_t, dense_t),
871 dense_t = constant_op.constant(dense_vals_np)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | csr_sparse_matrix_to_dense_op.cc | 85 Tensor dense_t(cpu_allocator(), DataTypeToEnum<T>::value, in Compute() local
90 set_zero(context->eigen_device<Device>(), dense_t.flat<T>()); in Compute()
92 auto dense_ptr = dense_t.flat<T>().data(); in Compute()
118 context->set_output(0, dense_t); in Compute()
193 Tensor dense_t; in Compute() local
201 c, indices_t, values_t, dense_tensor_shape, &dense_t, in Compute()
203 c->set_output(0, dense_t); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared.cc | 62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
66 OP_REQUIRES_OK(ctx, ctx->input("dense", &dense_t)); in Compute()
85 const auto rhs_dims = BCast::FromShape(dense_t->shape()); in Compute()
125 dense_t->shaped<T, NDIM>(b.y_reshape()) \ in Compute()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor_test.cc | 387 auto dense_t = dense.tensor<tstring, 3>(); in TEST() local
391 EXPECT_EQ(dense_t(ix_n), vals_t(n)); in TEST()
395 EXPECT_EQ(dense_t(0, 0, 1), ""); in TEST()
396 EXPECT_EQ(dense_t(0, 0, 3), ""); in TEST()
397 EXPECT_EQ(dense_t(3, 3, 3), ""); in TEST()
398 EXPECT_EQ(dense_t(3, 3, 4), ""); in TEST()
427 auto dense_t = dense.tensor<tstring, 3>(); in TEST() local
431 EXPECT_EQ(dense_t(ix_n), vals_t(n)); in TEST()
435 EXPECT_EQ(dense_t(0, 0, 1), ""); in TEST()
436 EXPECT_EQ(dense_t(0, 0, 3), ""); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 37 dense_t = ops.convert_to_tensor(dense)
38 locs = array_ops.stop_gradient(array_ops.where(math_ops.abs(dense_t) > 0))
39 return sparse_csr_matrix_ops.dense_to_csr_sparse_matrix(dense_t, locs)
|
| D | csr_sparse_matrix_grad_test.py | 36 dense_t = ops.convert_to_tensor(dense)
37 locs = array_ops.stop_gradient(array_ops.where(math_ops.abs(dense_t) > 0))
38 return sparse_csr_matrix_ops.dense_to_csr_sparse_matrix(dense_t, locs)
|
| D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 37 dense_t = ops.convert_to_tensor(dense)
38 locs = array_ops.stop_gradient(array_ops.where(math_ops.abs(dense_t) > 0))
39 return sparse_csr_matrix_ops.dense_to_csr_sparse_matrix(dense_t, locs)
|
| D | csr_sparse_matrix_ops_test.py | 54 dense_t = ops.convert_to_tensor(dense)
55 locs = array_ops.stop_gradient(array_ops.where(math_ops.abs(dense_t) > 0))
56 return sparse_csr_matrix_ops.dense_to_csr_sparse_matrix(dense_t, locs)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 792 def sparse_dense_cwise_add(sp_t, dense_t): argument
813 sp_t.dense_shape, dense_t)
|