Home
last modified time | relevance | path

Searched refs:dense_t (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_tensor_dense_matmul_op_test.py148 dense_t = np.matrix([[1] * 5, [2] * 5], dtype=np.float32)
151 self.evaluate(sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
152 dense_t = np.matrix([[1] * 500, [2] * 500], dtype=np.float32)
155 self.evaluate(sparse_ops.sparse_tensor_dense_matmul(sparse_t, dense_t))
158 dense_t = np.matrix([[1] * 5, [2] * 5, [3] * 5], dtype=np.float32)
163 sparse_t, dense_t, adjoint_a=True))
164 dense_t = np.matrix([[1] * 500, [2] * 500, [3] * 500], dtype=np.float32)
169 sparse_t, dense_t, adjoint_a=True))
183 dense_t = np.matrix([[1] * 5, [2] * 5], dtype=np.float32)
187 sparse_t, dense_t))
[all …]
Dsparse_tensor_dense_matmul_grad_test.py71 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,
Dsparse_ops_test.py811 dense_t = constant_op.constant(dense_vals_np)
813 self._check(sp_t / dense_t, sp_t_densified / dense_vals_np, sp_t)
815 self._check(sp_t * dense_t, sp_t_densified * dense_vals_np, sp_t)
816 self._check(dense_t * sp_t, sp_t_densified * dense_vals_np, sp_t)
819 res = sp_t / dense_t # should invoke "__truediv__"
830 dense_t = array_ops.ones(shape, dtype=dtypes.int32)
832 sparse_ops.sparse_dense_cwise_add(sp_t, dense_t),
836 dense_t = array_ops.ones([1], dtype=dtypes.int32)
838 sparse_ops.sparse_dense_cwise_add(sp_t, dense_t),
853 dense_t = constant_op.constant(dense_vals_np)
[all …]
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor_test.cc383 auto dense_t = dense.tensor<string, 3>(); in TEST() local
387 EXPECT_EQ(dense_t(ix_n), vals_t(n)); in TEST()
391 EXPECT_EQ(dense_t(0, 0, 1), ""); in TEST()
392 EXPECT_EQ(dense_t(0, 0, 3), ""); in TEST()
393 EXPECT_EQ(dense_t(3, 3, 3), ""); in TEST()
394 EXPECT_EQ(dense_t(3, 3, 4), ""); in TEST()
423 auto dense_t = dense.tensor<string, 3>(); in TEST() local
427 EXPECT_EQ(dense_t(ix_n), vals_t(n)); in TEST()
431 EXPECT_EQ(dense_t(0, 0, 1), ""); in TEST()
432 EXPECT_EQ(dense_t(0, 0, 3), ""); in TEST()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsparse_dense_binary_op_shared.cc62 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/python/ops/
Dsparse_ops.py622 def sparse_dense_cwise_add(sp_t, dense_t): argument
643 sp_t.dense_shape, dense_t)