| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | softmax_op.cc | 66 Tensor dense_shape_t = logits_matrix->dense_shape(); in Compute() local
71 DataTypeToEnum<T>::value, dense_shape_t, in Compute()
149 Tensor dense_shape_t = softmax_matrix->dense_shape(); in Compute() local
151 static_cast<const Tensor>(dense_shape_t).vec<int64>(); in Compute()
160 dense_shape_t.SummarizeValue(3), " vs. ", in Compute()
178 DataTypeToEnum<T>::value, dense_shape_t, in Compute()
|
| D | zeros_op.h | 44 const Tensor& dense_shape_t, CSRSparseMatrix* matrix) { in operator()
45 auto dense_shape = dense_shape_t.vec<int64>(); in operator()
78 dtype, dense_shape_t, batch_ptr_t, csr_row_ptr_t, coo_col_ind_t, in operator()
|
| D | csr_sparse_matrix_to_dense_op.cc | 66 const Tensor& dense_shape_t = csr_sparse_matrix->dense_shape(); in Compute() local
67 const int rank = dense_shape_t.dim_size(0); in Compute()
72 auto dense_shape = dense_shape_t.vec<int64>(); in Compute()
138 const Tensor& dense_shape_t = csr_sparse_matrix->dense_shape(); in Compute() local
139 const int rank = dense_shape_t.dim_size(0); in Compute()
147 auto dense_shape = dense_shape_t.vec<int64>(); in Compute()
|
| D | zeros_op.cc | 48 const Tensor& dense_shape_t = c->input(0); in Compute() local
52 csr_sparse_matrix_zeros(c, dtype_, dense_shape_t, &matrix)); in Compute()
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 127 const Tensor& dense_shape_t = c->input(2); in ComputeAsync() local
128 const int rank = dense_shape_t.NumElements(); in ComputeAsync()
134 auto dense_shape = dense_shape_t.vec<int64>(); in ComputeAsync()
177 &indices_t, &dense_shape_t, dense_shape, rows, cols, in ComputeAsync()
266 values_t.dtype(), dense_shape_t, batch_ptr_t, csr_row_ptr_t, in ComputeAsync()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 147 const Tensor& dense_shape_t = csr_sparse_matrix->dense_shape(); in Compute() local
148 c->set_output(2, dense_shape_t); in Compute()
149 const int rank = dense_shape_t.dim_size(0); in Compute()
153 auto dense_shape = dense_shape_t.vec<int64>(); in Compute()
|
| D | dense_to_csr_sparse_matrix_op.cc | 249 Tensor dense_shape_t(cpu_allocator(), DT_INT64, TensorShape({rank})); in ComputeAsync() local
250 auto dense_shape_mutable = dense_shape_t.vec<int64>(); in ComputeAsync()
254 auto dense_shape = const_cast<const Tensor&>(dense_shape_t).vec<int64>(); in ComputeAsync()
333 values_t.dtype(), dense_shape_t, batch_ptr_t, csr_row_ptr_t, in ComputeAsync()
|
| D | sparse_matrix.h | 457 auto dense_shape_t = dense_shape.vec<int64>(); in ValidateTypesAndShapes() local
458 const int64 batch_size = (rank == 2) ? 1 : dense_shape_t(0); in ValidateTypesAndShapes()
459 const int64 num_rows = (rank == 2) ? dense_shape_t(0) : dense_shape_t(1); in ValidateTypesAndShapes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 44 const Tensor& dense_shape_t) { in operator ()()
53 const auto dense_shape = dense_shape_t.vec<Tindex>(); in operator ()()
202 const Tensor& dense_shape_t = context->input(kDenseShapeInput); in Compute() local
205 OP_REQUIRES(context, TensorShapeUtils::IsVector(dense_shape_t.shape()), in Compute()
207 dense_shape_t.shape().DebugString())); in Compute()
222 dense_shape_t)); in Compute()
|
| D | sparse_fill_empty_rows_op.h | 31 const Tensor& dense_shape_t);
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops_test.cc | 37 Tensor dense_shape_t = test::AsTensor<int64>({5, 6}); in TEST() local
38 op.input_tensors[2] = &dense_shape_t; in TEST()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops_test.py | 97 dense_shape_t = constant_op.constant(dense_shape, dtype=dtypes.int64)
101 indices=indices_t, values=values_t, dense_shape=dense_shape_t)
|