Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_concat.h34 SparseConcat() : BaseOperator(kNameSparseConcat) { in SparseConcat() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_concat_cpu_kernel.cc121 bool SparseConcatCpuKernelMod::SparseConcat(const std::vector<kernel::KernelTensor *> &inputs, in SparseConcat() function in mindspore::kernel::SparseConcatCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dsparse_ops.py1039 class SparseConcat(Primitive): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini3318 [SparseConcat] section
3319 opInfo.engine=DNN_VM_AICPU
3320 opInfo.flagPartial=False
3321 opInfo.computeCost=100
3322 opInfo.flagAsync=False
3323 opInfo.opKernelLib=CUSTAICPUKernel
3324 opInfo.kernelSo=libcust_cpu_kernels.so
3325 opInfo.functionName=RunCpuKernel
3326 opInfo.workspaceSize=1024
3327 opInfo.opsFlag=OPS_FLAG_CLOSE
[all …]