| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/ |
| H A D | SparseCsrLinearAlgebra.cpp | 17 const Tensor& dense, in _sparse_mm_mkl_()
105 float* dense, in sparse_mm()
141 double* dense, in sparse_mm()
187 const Tensor& dense, in sparse_mm_mkl_template()
220 const Tensor& dense, in _sparse_mm_mkl_()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_dense_binary_op_shared_test.cc | 107 Tensor dense(DT_FLOAT, TensorShape(shape)); in TEST_F() local
135 Tensor dense(DT_FLOAT, TensorShape({3, 1})); in TEST_F() local
162 Tensor dense(DT_FLOAT, TensorShape({2})); in TEST_F() local
194 Tensor dense(DT_FLOAT, TensorShape({2})); in TEST_F() local
214 Node* sp_shape, Node* dense) { in SparseMatCMulDenseMat()
|
| H A D | sparse_to_dense_op_gpu.cu.cc | 40 const int ndims, T* __restrict__ dense) { in SparseToDenseKernel()
56 T* __restrict__ dense) { in SetDefaultValue()
115 const int num_dims, T* dense) { in LaunchComputeKernels()
145 const Tensor& shape, const T default_value, Tensor* dense) { in operator ()()
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | _triton_ops_meta.py | 502 def bench(meta, bsr=bsr, dense=dense): argument
571 def tune__int_bsr_dense_addmm( argument
597 def tune_bsr_dense_addmm( argument
660 def bench(meta, input=input, bsr=bsr, dense=dense, alpha=alpha, out=out): argument
845 def bench(meta, bsr=bsr, dense=dense): argument
|
| H A D | _semi_structured_conversions.py | 47 def sparse_semi_structured_from_dense_cutlass(dense): argument
285 def _sparse_semi_structured_tile(dense): argument
315 def _compute_compressed_swizzled_bitmask(dense): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCUDATensorMath.cu | 77 …beta, const Tensor& t, const Scalar& alpha, Tensor& indices, Tensor& values, const Tensor& dense) { in s_addmm_out_sparse_dense_cuda_worker()
90 …nse_cuda(Tensor& r_, const Tensor& t, const SparseTensor& sparse_, const Tensor& dense, const Scal… in s_addmm_out_sparse_dense_cuda()
144 const Tensor& dense, in s_addmm_sparse_dense_cuda()
167 const Tensor& dense, in s_addmm_sparse_dense_cuda_()
184 const Tensor& dense, in hspmm_out_sparse_cuda()
244 SparseTensor hspmm_sparse_cuda(const SparseTensor& sparse, const Tensor& dense) { in hspmm_sparse_cuda()
271 Tensor& add_out_dense_sparse_cuda(Tensor& r_, const Tensor& dense, const SparseTensor& sparse, cons… in add_out_dense_sparse_cuda()
|
| H A D | SparseBlasLegacy.cpp | 14 …const Tensor& crow_indices, const Tensor& col_indices, const Tensor& values, const Tensor& dense) { in s_addmm_out_csr_sparse_dense_cuda_worker()
|
| H A D | SparseSemiStructuredOps.cu | 847 _to_sparse_semi_structured(const Tensor& dense) { in _to_sparse_semi_structured()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_masked.py | 328 def to_sparse(dense): argument
337 def to_sparse(dense): argument
345 def to_sparse(dense): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensorMath.cpp | 710 Tensor& add_out_dense_sparse_cpu(Tensor& r, const Tensor& dense, const SparseTensor& sparse_, const… in add_out_dense_sparse_cpu()
1186 …Tensor& t, const Scalar& alpha, const Tensor& indices, const Tensor& values, const Tensor& dense) { in s_addmm_out_sparse_dense_worker()
1239 const Tensor& dense, in s_addmm_out_sparse_dense_cpu()
1326 const Tensor& dense, in s_addmm_sparse_dense_cpu()
1349 const Tensor& dense, in s_addmm_sparse_dense_cpu_()
1361 const Tensor& dense, in _sparse_addmm()
1390 const Tensor& dense, in _sparse_mm_out()
1406 SparseTensor& hspmm_out_sparse_cpu(const SparseTensor& sparse_, const Tensor& dense, SparseTensor& … in hspmm_out_sparse_cpu()
1475 SparseTensor hspmm_sparse_cpu(const SparseTensor& sparse, const Tensor& dense) { in hspmm_sparse_cpu()
1490 const Tensor& dense, in _sspaddmm_out_cpu()
|
| H A D | SparseCsrTensorMath.cpp | 527 const Tensor& dense, in addmm_out_sparse_csr_native_cpu()
724 const Tensor& dense, in addmm_sparse_compressed_dense()
813 const Tensor& dense, in add_out_dense_sparse_compressed_cpu()
|
| /aosp_15_r20/external/libopus/dnn/torch/weight-exchange/wexchange/tf/ |
| H A D | tf.py | 90 def dump_tf_dense_weights(where, dense, name='dense', scale=1/128, sparse=False, diagonal=False, qu… argument
110 def load_tf_dense_weights(path, dense): argument
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/aho-corasick-1.1.3/src/nfa/ |
| H A D | noncontiguous.rs | 127 dense: Vec<StateID>, field
388 let dense = self.states[prev].dense; in add_transition() localVariable
725 dense: StateID, field
1513 let dense = self.nfa.alloc_dense_state()?; in densify() localVariable
1623 let dense = start.dense; in close_start_state_loop_for_leftmost() localVariable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorBinaryOps.cu | 30 const T* dense, in op_dense_esuhm()
54 const T* dense, // [batch_size x embedding_dim] in nested_op_dense_kernelLauncher()
|
| /aosp_15_r20/external/libopus/dnn/torch/weight-exchange/wexchange/torch/ |
| H A D | torch.py | 249 def dump_torch_dense_weights(where, dense, name='dense', scale=1/128, sparse=False, diagonal=False,… argument
267 def load_torch_dense_weights(where, dense): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| H A D | custom_call_encoding.cc | 125 static mlir::DenseElementsAttr Flatten(DenseIntOrFPElementsAttr dense) { in Flatten()
142 DenseIntOrFPElementsAttr dense = value.cast<DenseIntOrFPElementsAttr>(); in PackDenseElementsAttribute() local
644 if (auto dense = attr.dyn_cast<DenseIntOrFPElementsAttr>()) in Match() local
651 auto dense = attr.cast<DenseIntOrFPElementsAttr>(); in Encode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/ |
| H A D | sparse_tensor_test.cc | 385 Tensor dense(DT_STRING, TensorShape({4, 4, 5})); in TEST() local
425 Tensor dense(DT_STRING, TensorShape({10, 10, 10})); in TEST() local
744 Tensor dense(DT_INT32, TensorShape({})); in TEST() local
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/regex-automata-0.4.6/tests/gen/ |
| H A D | mod.rs | 1 mod dense; module
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/regex-automata-0.4.6/tests/fuzz/ |
| H A D | mod.rs | 1 mod dense; module
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/layers/ |
| H A D | core.py | 24 dense = core.dense variable
|
| /aosp_15_r20/external/eigen/Eigen/src/SparseLU/ |
| H A D | SparseLU_kernel_bmod.h | 39 … LU_kernel_bmod<SegSizeAtCompileTime>::run(const Index segsize, BlockScalarVector& dense, ScalarVe… in run()
|
| H A D | SparseLU_copy_to_ucol.h | 51 … BlockIndexVector repfnz ,IndexVector& perm_r, BlockScalarVector dense, GlobalLU_t& glu) in copy_to_ucol()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| H A D | csr_sparse_matrix_grad_test.py | 31 def dense_to_csr_sparse_matrix(dense): argument
|
| /aosp_15_r20/external/rust/android-crates-io/crates/regex-automata/src/ |
| D | sparse_set.rs | 17 dense: Vec<usize>, field
|
| /aosp_15_r20/prebuilts/go/linux-x86/src/cmd/compile/internal/ssa/ |
| D | sparseset.go | 11 dense []ID member
|