Home
last modified time | relevance | path

Searched defs:dense_tensor (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_addmm.cc217 auto *dense_tensor = ctx.Input(3); in ComputeRowAndCol2() local
283 auto *dense_tensor = ctx.Input(3); in SparseAddmmCompute() local
/third_party/mindspore/mindspore-src/source/tests/st/sparse/
Dtest_csr.py437 def construct(self, indptr, indices, values, dense_tensor, dense_vector): argument
792 def test_to_csr(dense_tensor): argument
Dtest_coo.py403 def test_to_coo(dense_tensor): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_cross_cpu_kernel.cc298 std::vector<int64_t> dense_tensor = dense_list_in[i]; in GenerateColumnsFromInput() local