Home
last modified time | relevance | path

Searched refs:UnsortedSegmentMax (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dunsorted_segment_max_gpu_kernel.cc22 UnsortedSegmentMax,
27 UnsortedSegmentMax,
32 UnsortedSegmentMax,
37 UnsortedSegmentMax,
42 UnsortedSegmentMax,
47 UnsortedSegmentMax,
52 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax,
60 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax,
68 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax,
76 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax,
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dunsorted_segment_max.cu21 __global__ void UnsortedSegmentMax(const T *input, const S *segment_ids, const int64_t num_segments… in UnsortedSegmentMax() function
69 …UnsortedSegmentMax<<<GET_BLOCKS(size), GET_THREADS, 0, stream>>>(input, segment_ids, num_segments,… in CalUnsortedSegmentMax()
/third_party/mindspore/tests/st/ops/gpu/
Dtest_unsorted_segment_max.py31 self.unsorted_segment_max = P.UnsortedSegmentMax()
147 net = P.UnsortedSegmentMax()
212 self.unsorted_segment_max = P.UnsortedSegmentMax()
/third_party/mindspore/tests/ut/python/parallel/
Dtest_unsortedsegmentmax.py38 self.merge_op = P.UnsortedSegmentMax().shard((strategy1, strategy2))
/third_party/mindspore/mindspore/ops/operations/
D__init__.py33 … Transpose, TruncatedNormal, TupleToArray, UnsortedSegmentMin, UnsortedSegmentMax,
Darray_ops.py2174 class UnsortedSegmentMax(PrimitiveWithCheck): class
/third_party/mindspore/mindspore/nn/layer/
Dembedding.py503 self.merge_op = P.UnsortedSegmentMax()
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py938 @bprop_getters.register(P.UnsortedSegmentMax)
/third_party/mindspore/tests/ut/python/ops/
Dtest_ops.py1958 'block': P.UnsortedSegmentMax(),
/third_party/mindspore/config/
Dop_info.config436 {"op_name": "UnsortedSegmentMax", "inputs": [{"index": 0, "name": "data", "need_compile": false, "p…