Searched refs:dyn_a (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_unsorted_segment_sum.py | 113 def __init__(self, num_segments, dyn_a=True, dyn_b=True): argument
118 self.to_dyn_1 = dyn_a
|
| D | test_unsorted_segment_min.py | 210 def __init__(self, num_segments, dyn_a=True, dyn_b=True): argument
215 self.to_dyn_1 = dyn_a
|
| D | test_unsorted_segment_max.py | 210 def __init__(self, num_segments, dyn_a=True, dyn_b=True): argument
215 self.to_dyn_1 = dyn_a
|
| D | test_gatherV2_op.py | 945 def __init__(self, axis=0, dyn_a=True, dyn_b=True): argument
949 self.to_dyn_1 = dyn_a
|