Home
last modified time | relevance | path

Searched defs:dense_shape (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/utils/
H A Dsparsity_format_converter_test.cc27 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_matrix.h146 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()
H A Dcsr_sparse_matrix_to_dense_op.cc72 auto dense_shape = dense_shape_t.vec<int64_t>(); in Compute() local
148 auto dense_shape = dense_shape_t.vec<int64_t>(); in Compute() local
H A Dcsr_sparse_matrix_to_sparse_tensor_op.cc80 const Tensor& dense_shape = csr_sparse_matrix->dense_shape(); in Compute() local
154 auto dense_shape = dense_shape_t.vec<int64_t>(); in Compute() local
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc68 const Tensor& dense_shape = ctx->input(2); in Compute() local
150 auto dense_shape = dense_shape_t.vec<int64_t>(); in ComputeAsync() local
H A Ddense_to_csr_sparse_matrix_op.cc89 Tensor dense_shape(cpu_allocator(), DT_INT64, TensorShape({rank})); in Compute() local
254 auto dense_shape = in ComputeAsync() local
H A Dzeros_op.h45 auto dense_shape = dense_shape_t.vec<int64_t>(); in operator() local
H A Dsparse_ordering_amd_op.cc71 const Tensor& dense_shape = input_matrix->dense_shape(); in Compute() local
H A Dsparse_matrix_components_op.cc71 auto dense_shape = csr_sparse_matrix->dense_shape().vec<int64_t>(); in Compute() local
H A Dsparse_cholesky_op.cc240 const Tensor& dense_shape = sparse_matrix.dense_shape(); in ValidateInputs() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsparse_tensor_test.py152 def my_func(dense_shape): argument
204 def my_func(dense_shape): argument
281 def testToFromComponents(self, st_spec, indices, values, dense_shape): argument
336 def testToFromTensorList(self, st_spec, indices, values, dense_shape): argument
H A Dindexed_slices.py109 def __init__(self, values, indices, dense_shape=None): argument
126 def dense_shape(self): member in IndexedSlices
H A Dsparse_tensor.py112 def __init__(self, indices, values, dense_shape): argument
213 def dense_shape(self): member in SparseTensor
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_csr_matrix_ops.cc69 ShapeHandle dense_shape; in __anona88575120102() local
104 ShapeHandle dense_shape = c->Vector(rank); in __anona88575120202() local
117 ShapeHandle dense_shape = c->input(0); in __anona88575120302() local
459 ShapeHandle dense_shape; in __anona88575120b02() local
H A Dparsing_ops.cc35 for (const auto& dense_shape : dense_shapes) { in AddDenseOutputShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dsparse_tensor_slice_dataset_op.cc76 std::vector<int64_t> dense_shape; in AsGraphDefInternal() local
239 const Tensor* dense_shape; in MakeDataset() local
H A Dsparse_tensor_slice_dataset_op_test.cc33 Tensor dense_shape, DataType tvalues, in SparseTensorSliceDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/
H A Dcross_device_ops_test.py96 def _make_indexed_slices(values, indices, dense_shape, device): argument
105 def _make_mirrored_indexed_slices(devices, values, indices, dense_shape): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dsparse_to_dense.py38 def generate_index(dense_shape): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_split_op.cc122 TensorShape dense_shape; in SparseSplitOpImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_concat_op_test.py107 def _SparseTensor_NoNonZeros(self, dense_shape): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dembedding_lookup_sparse.cc156 const TfLiteTensor* dense_shape; in Eval() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Ddense_to_sparse_batch_dataset_op.cc179 Tensor dense_shape(ctx->allocator({}), DT_INT64, {row_ndims + 1}); in GetNextInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistributed_variable_test.py615 def _make_index_slices(values, indices, dense_shape=None): argument
H A Dvalues_v2_test.py471 def _make_index_slices(values, indices, dense_shape=None): argument

12