| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_segment_op.py | 33 def construct(self, x, segment_ids): argument
43 def construct(self, x, segment_ids): argument
53 def construct(self, x, segment_ids): argument
63 def construct(self, x, segment_ids): argument
73 def construct(self, x, segment_ids): argument
|
| D | test_sparse_segment_mean_op.py | 30 def construct(self, x, indices, segment_ids): argument
34 def sparse_segment_mean_numpy(x, indices, segment_ids): argument
|
| D | test_sparse_segment_mean_with_num_segments_op.py | 30 def construct(self, x, indices, segment_ids, num_segments): argument
34 def sparse_segment_mean_with_num_segments_numpy(x, indices, segment_ids, num_segments): argument
|
| D | test_sparse_segment_sum_grad_op.py | 30 def construct(self, grad, indices, segment_ids, output_dim0): argument
|
| D | test_sparse_segment_sum_with_num_segments_op.py | 30 def construct(self, x, indices, segment_ids, num_segments): argument
|
| D | test_sparse_segment_mean_grad_op.py | 30 def construct(self, grad, indices, segment_ids, output_dim0): argument
|
| D | test_sparse_segment_sqrt_n_grad_op.py | 30 def construct(self, grad, indices, segment_ids, output_dim0): argument
|
| D | test_sparse_segment_sqrt_n_op.py | 32 def construct(self, x, indices, segment_ids): argument
|
| D | test_unsorted_segment_prod.py | 60 def unsorted_segment_arith_expected(func, x, segment_ids, num_segments): argument
408 def prod_vmap_graph(x, segment_ids, num_segments): argument
|
| D | test_sparse_segment_sum_op.py | 32 def construct(self, x, indices, segment_ids): argument
|
| D | test_sparse_segment_sqrt_n_with_num_segments_op.py | 32 def construct(self, x, indices, segment_ids, num_segments): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_unsorted_segment_arithmetic.py | 145 def unsorted_segment_arith_expected(func, x, segment_ids, num_segments): argument
304 def min_vmap_graph(x, segment_ids, num_segments): argument
308 def max_vmap_graph(x, segment_ids, num_segments): argument
312 def sum_vmap_graph(x, segment_ids, num_segments): argument
|
| D | test_sparse_segment_mean_op.py | 30 def construct(self, x, indices, segment_ids): argument
34 def sparse_segment_mean_numpy(x, indices, segment_ids): argument
|
| D | test_sparse_segment_mean_with_num_segments_op.py | 31 def construct(self, x, indices, segment_ids, num_segments): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/ |
| D | test_unsorted_segment_sum_dyn.py | 29 def construct(self, x, segment_ids, num_segments): argument
39 def construct(self, x, segment_ids): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_unsorted_segment_op.py | 50 def unsorted_segment_arith_expected(func, x, segment_ids, num_segments): argument
125 def construct(self, x, segment_ids): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/ |
| D | grad_sparse_ops.py | 188 def bprop(x, indices, segment_ids, out, dout): argument
206 def bprop(x, indices, segment_ids, num_segments, out, dout): argument
225 def bprop_gpu(x, indices, segment_ids, out, dout): argument
235 def bprop(x, indices, segment_ids, out, dout): argument
271 def bprop_gpu(x, indices, segment_ids, num_segments, out, dout): argument
281 def bprop(x, indices, segment_ids, num_segments, out, dout): argument
305 def bprop(x, indices, segment_ids, num_segments, out, dout): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_unsortedsegmentsum.py | 26 def forward_func(input_x, segment_ids, num_segments): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_segment_mean.py | 15 def construct(self, x, segment_ids): argument
|
| D | test_sparse_segment_sqrt_n.py | 32 def construct(self, x, indices, segment_ids): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/fusion/ |
| D | test_unsorted_segment_sum_fission.py | 32 def construct(self, x, segment_ids): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/testTFBert5Rows2/ |
| D | datasetSchema.json | 10 "segment_ids" : { object
|
| /third_party/mindspore/mindspore-src/source/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/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | segment_min_proto.cc | 44 Tensor segment_ids; in IMPLEMT_COMMON_INFERFUNC() local
|
| D | segment_mean_proto.cc | 46 Tensor segment_ids; in IMPLEMT_COMMON_INFERFUNC() local
|