| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_sparse_to_dense_op.py | 35 def construct(self, indices, values, sparse_shape): argument
46 def construct(self, indices, values, sparse_shape, sense): argument
|
| D | test_sparse_addmm_op.py | 31 def construct(self, indices, values, sparse_shape, x2_dense, x3_dense, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | csr_elementwise.cc | 59 ShapeVector sparse_shape = ConvertToShapeVector(shape); in CSRElementWiseInfer() local
|
| D | csr_gather.cc | 50 auto sparse_shape = abstract::CheckArg<AbstractTuple>(op_name, input_args, kIndexThree); in CSRGatherInfer() local
|
| D | csr_reducesum.cc | 63 ShapeVector sparse_shape = ConvertToShapeVector(shape); in CSRReduceSumInfer() local
|
| D | csr_mm.cc | 61 ShapeVector sparse_shape = ConvertToShapeVector(shape); in CSRMMInfer() local
|
| D | csr_mv.cc | 60 ShapeVector sparse_shape = ConvertToShapeVector(shape); in CSRMVInfer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/sparse/ |
| D | sparse.py | 146 def construct(self, indices, values, sparse_shape, dense): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_dense_cwise_mul_cpu_kernel.cc | 69 std::vector<int64_t> sparse_shape(dimension); in ComputeMul() local
159 std::vector<int64_t> sparse_shape(dimension); in SparseDenseCwiseMulBcastCompute() local
|
| D | sparse_dense_cwise_div_cpu_kernel.cc | 71 std::vector<int64_t> sparse_shape(dimension); in ComputeDiv() local
175 std::vector<int64_t> sparse_shape(dimension); in SparseDenseCwiseDivBcastCompute() local
|
| D | sparse_dense_cwise_add_cpu_kernel.cc | 69 std::vector<int64_t> sparse_shape(dimension); in ComputeAdd() local
159 std::vector<int64_t> sparse_shape(dimension); in SparseDenseCwiseAddBcastCompute() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_sparse_tensor_dense_matmul_op.py | 39 def construct(self, indices, values, sparse_shape, dense): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_dense_cwise_utils.cc | 277 std::vector<int64_t> sparse_shape(dimension); in SparseDenseCwiseOpBcastCompute() local
409 std::vector<int64_t> sparse_shape(dimension); in ComputeOp() local
443 std::vector<int64_t> sparse_shape(dimension); in ComputeOpComplex() local
646 std::vector<int64_t> sparse_shape(dimension); in SparseDenseCwiseOpBcastComputeComplex() local
|
| D | setsize.cc | 108 auto sparse_shape = set_shape_->GetTensorShape(); in SparseTensorFromContext() local
|
| D | sparse_addmm.cc | 115 auto sparse_shape = shape_tensor->GetTensorShape(); in SparseAddmmCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/ |
| D | sparse_dense_cwise_operation_gpu_kernel.cc | 186 std::vector<int64_t> sparse_shape(dimension_); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/ |
| D | common_shape_fns.cc | 763 Shape sparse_shape = shape.GetShape(); in ValidateSparseTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/ |
| D | prim.cc | 853 AbstractBasePtrList sparse_shape = arg->shape()->elements(); in ConvertAbstractToPython() local
868 AbstractBasePtrList sparse_shape = arg->shape()->elements(); in ConvertAbstractToPython() local
|