Home
last modified time | relevance | path

Searched defs:dense (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/sparse/
Dtest_coo.py448 def test_coo_add_dense(indices, values, shape, dense): argument
452 def test_dense_add_coo(indices, values, shape, dense): argument
461 def test_coo_sub_dense(indices, values, shape, dense): argument
465 def test_dense_sub_coo(indices, values, shape, dense): argument
469 def test_coo_mul_dense(indices, values, shape, dense): argument
473 def test_dense_mul_coo(indices, values, shape, dense): argument
477 def test_coo_div_dense(indices, values, shape, dense): argument
Dtest_csr.py568 def test_csr_mul(indptr, indices, values, shape, dense): argument
574 def test_csr_div(indptr, indices, values, shape, dense): argument
586 def test_csrmv(indptr, indices, values, shape, dense): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/
Dtest_sparsetensordensematmul.py32 def construct(self, indices, values, dens_shape, dense): argument
43 def construct(self, indices, values, dens_shape, dense): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_sparse_tensor_dense_matmul_op.py39 def construct(self, indices, values, sparse_shape, dense): argument
50 def construct(self, indices, values, dens_shape, dense): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_sparse_tensor_dense_matmul_op.py34 def construct(self, indices, values, dens_shape, dense): argument
45 def construct(self, indices, values, dens_shape, dense): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dcsr_elementwise.cc49 auto dense = abstract::CheckArg<AbstractTensor>(op_name, input_args, 4); in CSRElementWiseInfer() local
Dcsr_gather.cc49 auto dense = abstract::CheckArg<AbstractTensor>(op_name, input_args, kIndexTwo); in CSRGatherInfer() local
Dcsr_mm.cc51 auto dense = abstract::CheckArg<AbstractTensor>(op_name, input_args, 4); in CSRMMInfer() local
Dcsr_mv.cc50 auto dense = abstract::CheckArg<AbstractTensor>(op_name, input_args, 4); in CSRMVInfer() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/sparse/
Dsparse.py146 def construct(self, indices, values, sparse_shape, dense): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/
Dgrad_sparse.py109 def bprop(indices, values, dense_shape, dense, out, dout): argument
213 def bprop(indptr, indices, values, dense_shape, dense, out, dout): argument
243 def bprop(indptr, indices, values, shape, dense, out, dout): argument
271 def bprop(indptr, indices, values, shape, dense, out, dout): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_addmm.cc215 …CpuKernelContext &ctx, Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> *dense, i… in ComputeRowAndCol2()
243 …const Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> &dense, int64_t row_x1, in… in ComputeRowAndCol3()
324 Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> dense(row_x2, col_x2); in SparseAddmmCompute() local
/third_party/typescript/src/services/
Dclassifier.ts258 const dense = classifications.spans; constant
587 const dense = classifications.spans; constant
Dclassifier2020.ts22 const dense = classifications.spans; constant
/third_party/rust/crates/unicode-ident/generate/src/
Dmain.rs44 let mut dense = Vec::<[u8; CHUNK]>::new(); in main() localVariable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dsparse_tensor_dense_matmul_helper.h72 S *dense = nullptr; in Process() local
/third_party/rust/crates/regex/src/
Dsparse.rs19 dense: Vec<usize>, field
Dcompile.rs1099 dense: Vec<SuffixCacheEntry>, field
/third_party/mesa3d/src/panfrost/bifrost/
Dnodearray.h67 nodearray_value *dense; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dgrad_sparse_ops.cc180 …MulDivCommon(BpropBuilder *ib, const NodePtr &indices, const NodePtr &indptr, const NodePtr &dense, in BpropCSRMulDivCommon()
/third_party/rust/crates/regex/src/literal/
Dimp.rs279 dense: Vec<u8>, field
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dmulti_head_attention_fusion.cc158 auto dense = VectorRef({dense_, input, weight, bias}); in DefineEmbedding() local
193 auto dense = VectorRef({dense_, input, weight}); in DefineEmbedding() local
610 bool IsTransposeWeight(const FuncGraphPtr &func_graph, const EquivPtr &equiv, VarPtr dense) { in IsTransposeWeight()