Searched refs:axis7 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_prod_op.py | 56 axis7 = (-2, -1) variable
98 self.axis7 = axis7
114 P.ReduceProd(self.keep_dims7)(self.x7, self.axis7),
162 expect7 = np.prod(x7, axis=axis7, keepdims=keep_dims7)
|
| D | test_reduce_min_op.py | 58 axis7 = (-2, -1) variable
100 self.axis7 = axis7
116 P.ReduceMin(self.keep_dims7)(self.x7, self.axis7),
170 expect7 = np.min(x7, axis=axis7, keepdims=keep_dims7)
|
| D | test_reduce_max_op.py | 58 axis7 = (-2, -1) variable
100 self.axis7 = axis7
116 P.ReduceMax(self.keep_dims7)(self.x7, self.axis7),
170 expect7 = np.max(x7, axis=axis7, keepdims=keep_dims7)
|
| D | test_reduce_sum_op.py | 57 axis7 = (1, 2) variable
123 self.axis7 = axis7
163 P.ReduceSum(self.keep_dims7)(self.x7, self.axis7),
223 expect7 = np.sum(x7, axis=axis7, keepdims=keep_dims7)
|
| D | test_reduce_mean_op.py | 55 axis7 = (1, 2) variable
121 self.axis7 = axis7
161 P.ReduceMean(self.keep_dims7)(self.x7, self.axis7),
221 expect7 = np.mean(x7, axis=axis7, keepdims=keep_dims7)
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_cumsum_op.py | 66 self.axis7 = axis0
136 self.cumsum_op(self.x7, self.axis7),
|