Home
last modified time | relevance | path

Searched defs:CSRSparseMatrixToSparseTensor (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dcsr_sparse_matrix_to_sparse_tensor.h31 CSRSparseMatrixToSparseTensor() : BaseOperator(kNameCSRSparseMatrixToSparseTensor) { in CSRSparseMatrixToSparseTensor() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dsparse_ops_proto.cc24 CUST_IMPLEMT_INFERFUNC(CSRSparseMatrixToSparseTensor, CSRSparseMatrixToSparseTensorInfer) { in CUST_IMPLEMT_INFERFUNC() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dcsr_sparse_matrix_to_sparse_tensor.cc28 const char *CSRSparseMatrixToSparseTensor = "CSRSparseMatrixToSparseTensor"; variable
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dsparse_ops.py675 class CSRSparseMatrixToSparseTensor(Primitive): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini831 [CSRSparseMatrixToSparseTensor] section
832 opInfo.subTypeOfInferShape=3
833 opInfo.opsFlag=OPS_FLAG_CLOSE
834 opInfo.engine=DNN_VM_AICPU
835 opInfo.flagPartial=False
836 opInfo.computeCost=100
837 opInfo.flagAsync=False
838 opInfo.opKernelLib=CUSTAICPUKernel
839 opInfo.formatAgnostic=False
840 opInfo.kernelSo=libcust_cpu_kernels.so
[all …]