Searched defs:UnsortedSegmentSum (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | unsorted_segment_sum.cu | 21 __global__ void UnsortedSegmentSum(size_t input_dim0, size_t input_dim1, size_t output_dim0, size_t… in UnsortedSegmentSum() function
38 void UnsortedSegmentSum(size_t input_dim0, size_t input_dim1, size_t output_dim0, size_t output_dim… in UnsortedSegmentSum() function
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | unsorted_segment_sum.h | 36 UnsortedSegmentSum() : PrimitiveC(kNameUnsortedSegmentSum) { in UnsortedSegmentSum() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/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/ops/operations/ |
| D | array_ops.py | 1993 class UnsortedSegmentSum(PrimitiveWithInfer): class
|