| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter_test.cc | 27 const std::vector<int> dense_shape = {3, 4}; in TEST() local
56 const std::vector<int> dense_shape = {3, 4}; in TEST() local
87 const std::vector<int> dense_shape = {3, 4}; in TEST() local
118 const std::vector<int> dense_shape = {3, 4}; in TEST() local
151 const std::vector<int> dense_shape = {3, 4}; in TEST() local
180 const std::vector<int> dense_shape = {3, 4}; in TEST() local
211 const std::vector<int> dense_shape = {3, 4}; in TEST() local
242 const std::vector<int> dense_shape = {3, 4}; in TEST() local
275 const std::vector<int> dense_shape = {3, 2, 2}; variable
311 const std::vector<int> dense_shape = {3, 2, 2}; variable
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | sparse_tensor_test.py | 154 def my_func(dense_shape): argument
206 def my_func(dense_shape): argument
283 def testToFromComponents(self, st_spec, indices, values, dense_shape): argument
327 def testToFromTensorList(self, st_spec, indices, values, dense_shape): argument
|
| D | indexed_slices.py | 101 def __init__(self, values, indices, dense_shape=None): argument
118 def dense_shape(self): member in IndexedSlices
|
| D | sparse_tensor.py | 116 def __init__(self, indices, values, dense_shape): argument
217 def dense_shape(self): member in SparseTensor
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.h | 146 const Tensor& dense_shape, // on host in CreateCSRSparseMatrix()
209 inline Tensor& dense_shape() { in dense_shape() function
214 inline const Tensor& dense_shape() const { in dense_shape() function
338 Tensor dense_shape = p.tensors_[0]; in Decode() local
400 CSRSparseMatrix(DataType dtype, const Tensor& dense_shape, in CSRSparseMatrix()
425 const Tensor& dense_shape, in ValidateTypesAndShapes()
|
| D | zeros_op.h | 45 auto dense_shape = dense_shape_t.vec<int64>(); in operator() local
|
| D | csr_sparse_matrix_to_dense_op.cc | 72 auto dense_shape = dense_shape_t.vec<int64>(); in Compute() local
147 auto dense_shape = dense_shape_t.vec<int64>(); in Compute() local
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 80 const Tensor& dense_shape = csr_sparse_matrix->dense_shape(); in Compute() local
153 auto dense_shape = dense_shape_t.vec<int64>(); in Compute() local
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 68 const Tensor& dense_shape = ctx->input(2); in Compute() local
134 auto dense_shape = dense_shape_t.vec<int64>(); in ComputeAsync() local
|
| D | dense_to_csr_sparse_matrix_op.cc | 89 Tensor dense_shape(cpu_allocator(), DT_INT64, TensorShape({rank})); in Compute() local
254 auto dense_shape = const_cast<const Tensor&>(dense_shape_t).vec<int64>(); in ComputeAsync() local
|
| D | sparse_ordering_amd_op.cc | 71 const Tensor& dense_shape = input_matrix->dense_shape(); in Compute() local
|
| D | sparse_matrix_components_op.cc | 71 auto dense_shape = csr_sparse_matrix->dense_shape().vec<int64>(); in Compute() local
|
| D | sparse_cholesky_op.cc | 239 const Tensor& dense_shape = sparse_matrix.dense_shape(); in ValidateInputs() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 69 ShapeHandle dense_shape; in __anon59a9d6c30102() local
104 ShapeHandle dense_shape = c->Vector(rank); in __anon59a9d6c30202() local
117 ShapeHandle dense_shape = c->input(0); in __anon59a9d6c30302() local
459 ShapeHandle dense_shape; in __anon59a9d6c30b02() local
|
| D | parsing_ops.cc | 35 for (const auto& dense_shape : dense_shapes) { in AddDenseOutputShapes() local
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | sparse_to_dense.py | 42 def generate_index(dense_shape): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | sparse_tensor_slice_dataset_op.cc | 74 std::vector<int64> dense_shape; in AsGraphDefInternal() local
237 const Tensor* dense_shape; in MakeDataset() local
|
| D | sparse_tensor_slice_dataset_op_test.cc | 29 Tensor dense_shape, DataType tvalues, in SparseTensorSliceDatasetParams()
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 99 def _make_indexed_slices(values, indices, dense_shape, device): argument
108 def _make_mirrored_indexed_slices(devices, values, indices, dense_shape): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse.cc | 155 const TfLiteTensor* dense_shape; in Eval() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_common_test.py | 166 def _make_indexed_slices(values, indices, dense_shape): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 179 Tensor dense_shape(ctx->allocator({}), DT_INT64, {row_ndims + 1}); in GetNextInternal() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 53 const auto dense_shape = dense_shape_t.vec<Tindex>(); in operator ()() local
|
| D | count_ops.cc | 68 Tensor* dense_shape; in OutputSparse() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | unicode_decode_op_test.py | 86 def _make_sparse_tensor(indices, values, dense_shape, dtype=np.int32): argument
|