| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | unsorted_segment_max.cu | 21 __global__ void UnsortedSegmentMax(const T *input, const S *segment_ids, const int64_t num_segments… in UnsortedSegmentMax()
60 void CalUnsortedSegmentMax(const T *input, const S *segment_ids, const int64_t num_segments, size_t… in CalUnsortedSegmentMax()
|
| D | unsorted_segment_min.cu | 31 __global__ void UnsortedSegmentMin(const T *input, const int *segment_ids, const int64_t num_segmen… in UnsortedSegmentMin()
65 void CalUnsortedSegmentMin(const T *input, const int *segment_ids, const int64_t num_segments, size… in CalUnsortedSegmentMin()
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_unsorted_segment_sum_fission.py | 32 def construct(self, x, segment_ids): argument
|
| /third_party/mindspore/tests/ut/data/dataset/testTFBert5Rows2/ |
| D | datasetSchema.json | 10 "segment_ids" : { object
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_unsorted_segment_sum.cc | 37 …auto segment_ids = TensorConstructUtils::CreateOnesTensor(kNumberTypeInt32, std::vector<int64_t>{1… in TEST_F() local
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_unsorted_segment_sum.py | 31 def construct(self, x, segment_ids, num_segments): argument
|
| /third_party/mindspore/tests/ut/data/dataset/testTFBert5Rows/ |
| D | datasetSchema.json | 35 "segment_ids" : { object
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_unsortedsegmentmax.py | 41 def construct(self, vectors, segment_ids): argument
|
| D | test_unsortedsegmentmin.py | 41 def construct(self, vectors, segment_ids): argument
|
| D | test_unsortedsegmentsum.py | 40 def construct(self, vectors, segment_ids): argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | dataset.py | 28 def __init__(self, input_ids, input_mask, segment_ids, label_id, seq_length=None): argument
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 897 def _unsorted_segment_min_or_max_grad(x, segment_ids, num_segments, out, dout): argument
921 def bprop(x, segment_ids, num_segments, out, dout): argument
932 def bprop(x, segment_ids, num_segments, out, dout): argument
942 def bprop(x, segment_ids, num_segments, out, dout): argument
960 def bprop(x, segment_ids, num_segments, out, dout): argument
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_arrays.cc | 234 auto segment_ids = CheckArg<AbstractTensor>(op_name, args_spec_list, 1); in InferImplUnsortedSegmentSum() local
289 auto segment_ids = CheckArg<AbstractTensor>(op_name, args_spec_list, 1); in InferImplUnsortedSegmentMax() local
341 auto segment_ids = CheckArg<AbstractTensor>(op_name, args_spec_list, 1); in InferImplUnsortedSegmentMin() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 2047 def __infer__(self, x, segment_ids, num_segments): argument
2156 def __check__(self, x, segment_ids, num_segments): argument
2282 def __check__(self, x, segment_ids, num_segments): argument
2343 def __infer__(self, x, segment_ids, num_segments): argument
|