Searched refs:a_dense_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 109 a_dense_shape = np.asarray([5, 6], dtype=np.int64)
112 shape=a_dense_shape)
121 dense_shape_proto = tensor_util.make_tensor_proto(a_dense_shape)
144 a_dense_shape = [5, 6]
147 shape=a_dense_shape)
151 a_st = sparse_tensor.SparseTensor(a_indices, a_values, a_dense_shape)
175 self.assertAllEqual(a_dense_shape, a_st_rt_value.dense_shape)
209 a_dense_shape = [3, 5, 6]
212 shape=a_dense_shape[1:]),
213 sparse.coo_matrix(([], ([], [])), shape=a_dense_shape[1:]),
[all …]
|
| D | csr_sparse_matrix_test.py | 46 a_dense_shape = [5, 6]
48 a_st = sparse_tensor.SparseTensor(a_indices, a_values, a_dense_shape)
51 self.assertEqual(a_sm.shape, a_dense_shape)
58 self.assertAllEqual(a_dense_shape, a_st_rt.dense_shape)
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | add_op.cc | 85 auto a_dense_shape = a.dense_shape().vec<int64>(); in operator ()() local
89 const int64 rows = a_dense_shape((rank == 2) ? 0 : 1); in operator ()()
113 a.values_vec<T>(i), a_dense_shape}; in operator ()()
138 a.values_vec<T>(i), a_dense_shape}; in operator ()()
168 a.values_vec<T>(i), a_dense_shape}; in operator ()()
|
| D | mul_op.cc | 59 auto a_dense_shape = a_dense_shape_t.vec<int64>(); in Compute() local
60 const int batch_size = a_dense_shape(0); in Compute()
|
| D | mat_mul_op.cc | 126 const auto& a_dense_shape = sparse_matrix_a.dense_shape().vec<int64>(); in ValidateInputs() local
128 a_dense_shape(this->transpose_a_ ? *rank - 2 : *rank - 1); in ValidateInputs()
134 TensorShape(a_dense_shape), " vs. ", in ValidateInputs()
511 auto a_dense_shape = a_dense_shape_t.vec<int64>(); in Compute() local
513 ctx, TensorShapeUtils::MakeShape(a_dense_shape, &a_dense_tensor_shape)); in Compute()
549 TTypes<int64>::ConstVec a_dense_shape_comp(a_dense_shape.data() + row_dim, in Compute()
|
| D | sparse_mat_mul_op.cc | 332 auto a_dense_shape = a_matrix->dense_shape().vec<int64>(); in Compute() local
338 TensorShapeUtils::MakeShape(a_dense_shape, &a_tensor_shape)); in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseMatrixOrderingAMD.pbtxt | 44 a_dense_shape = [4, 4]
48 a_st = tf.sparse.SparseTensor(a_indices, a_values, a_dense_shape)
|
| D | api_def_SparseMatrixSparseCholesky.pbtxt | 57 a_dense_shape = [4, 4]
61 a_st = tf.sparse.SparseTensor(a_indices, a_values, a_dense_shape)
|
| D | api_def_SparseMatrixSparseMatMul.pbtxt | 66 a_dense_shape = [4, 5]
74 a_st = tf.sparse.SparseTensor(a_indices, a_values, a_dense_shape)
|