Searched defs:UnsortedSegmentSum (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/ |
| D | unsorted_segment_sum.cc | 21 class MS_CORE_API UnsortedSegmentSum : public InferShapeOp { class
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | unsorted_segment_sum_base.h | 25 #define UnsortedSegmentSum(type, type1, input, unit_num, input_dim1, indices, output, output_dim0, … macro
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | unsorted_segment_sum.cu | 46 cudaError_t UnsortedSegmentSum(size_t input_dim0, size_t input_dim1, size_t output_dim0, size_t out… in UnsortedSegmentSum() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/expander/core/ |
| D | emitter.h | 155 …NodePtr UnsortedSegmentSum(const NodePtr &x, const NodePtr &segment_ids, const NodePtr &num_segmen… in UnsortedSegmentSum() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/ |
| D | gen_lite_ops.h | 2723 UnsortedSegmentSum() : BaseOperator(kNameUnsortedSegmentSum) {} in UnsortedSegmentSum() function
|
| /third_party/mindspore/mindspore-src/source/config/ |
| D | super_bar_config.json | 462 "UnsortedSegmentSum ": " check support failed when shape is -2", string
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/ |
| D | cust_aicpu_kernel.ini | 5306 [UnsortedSegmentSum] section
5307 opInfo.subTypeOfInferShape = 1
5308 opInfo.opsFlag = OPS_FLAG_CLOSE
5309 opInfo.engine = DNN_VM_AICPU
5310 opInfo.flagPartial = False
5311 opInfo.computeCost = 100
5312 opInfo.flagAsync = False
5313 opInfo.opKernelLib = CUSTAICPUKernel
5314 opInfo.formatAgnostic = False
5315 opInfo.userDefined = True
[all …]
|