| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter_test.cc | 27 const std::vector<int> dense_shape = {3, 4}; in TEST() local
51 const std::vector<int> dense_shape = {3, 4}; in TEST() local
77 const std::vector<int> dense_shape = {3, 4}; in TEST() local
103 const std::vector<int> dense_shape = {3, 4}; in TEST() local
131 const std::vector<int> dense_shape = {3, 4}; in TEST() local
155 const std::vector<int> dense_shape = {3, 4}; in TEST() local
181 const std::vector<int> dense_shape = {3, 4}; in TEST() local
207 const std::vector<int> dense_shape = {3, 4}; in TEST() local
236 const std::vector<int> dense_shape = {3, 4, 2}; variable
269 const std::vector<int> dense_shape = {3, 2, 4}; variable
[all …]
|
| /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_ordering_amd_op.cc | 71 const Tensor& dense_shape = input_matrix->dense_shape(); in Compute() local
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 67 const Tensor& dense_shape = ctx->input(2); in Compute() local
133 auto dense_shape = dense_shape_t.vec<int64>(); in ComputeAsync() local
|
| D | dense_to_csr_sparse_matrix_op.cc | 88 Tensor dense_shape(cpu_allocator(), DT_INT64, TensorShape({rank})); in Compute() local
253 auto dense_shape = const_cast<const Tensor&>(dense_shape_t).vec<int64>(); in ComputeAsync() 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/lite/testing/op_tests/ |
| D | sparse_to_dense.py | 42 def generate_index(dense_shape): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 69 ShapeHandle dense_shape; in __anond96a6a670102() local
104 ShapeHandle dense_shape = c->Vector(rank); in __anond96a6a670202() local
117 ShapeHandle dense_shape = c->input(0); in __anond96a6a670302() local
459 ShapeHandle dense_shape; in __anond96a6a670b02() local
|
| D | parsing_ops.cc | 35 for (const auto& dense_shape : dense_shapes) { in AddDenseOutputShapes() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | sparse_tensor_test.py | 180 def testToFromComponents(self, st_spec, indices, values, dense_shape): argument
224 def testToFromTensorList(self, st_spec, indices, values, dense_shape): argument
|
| D | indexed_slices.py | 91 def __init__(self, values, indices, dense_shape=None): argument
108 def dense_shape(self): member in IndexedSlices
|
| D | sparse_tensor.py | 117 def __init__(self, indices, values, dense_shape): argument
185 def dense_shape(self): member in SparseTensor
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | sparse_tensor_slice_dataset_op.cc | 70 std::vector<int64> dense_shape; in AsGraphDefInternal() local
232 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/ |
| D | cross_device_ops_test.py | 89 def _make_indexed_slices(values, indices, dense_shape, device): argument
98 def _make_mirrored_indexed_slices(devices, values, indices, dense_shape): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse.cc | 145 const TfLiteTensor* dense_shape = GetInput(context, node, 2); in Eval() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 78 const auto dense_shape = dense_shape_t.vec<int64>(); in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 173 Tensor dense_shape(ctx->allocator({}), DT_INT64, {row_ndims + 1}); in GetNextInternal() local
|
| D | parse_example_dataset_op.cc | 66 TensorShape dense_shape; in ParseExampleDatasetOp() 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
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.cc | 653 TensorShape dense_shape; in GetDenseShapes() local
|