Searched refs:axis5 (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_cumsum_op.py | 32 axis5 = -1 variable
43 list2 = [axis0, axis1, axis2, axis3, axis4, axis5, axis6]
61 self.axis5 = axis5
76 self.axis12 = axis5
91 self.axis19 = axis5
106 self.axis26 = axis5
121 self.axis33 = axis5
134 self.cumsum_op(self.x5, self.axis5),
|
| D | test_reduce_op.py | 35 self.axis5 = (-1,)
53 self.reduce_max(indice, self.axis5),
60 self.reduce_min(indice, self.axis5),
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_cumsum_op.py | 43 axis5 = 1
68 self.axis5 = axis5
80 P.CumSum()(self.x5, self.axis5),
117 expect5 = np.cumsum(x5, axis=axis5)
|
| D | test_cumprod_op.py | 43 axis5 = 1
68 self.axis5 = axis5
80 P.CumProd()(self.x5, self.axis5),
117 expect5 = np.cumprod(x5, axis=axis5)
|
| D | test_reduce_prod_op.py | 47 axis5 = () variable
90 self.axis5 = axis5
112 P.ReduceProd(self.keep_dims5)(self.x5, self.axis5),
|
| D | test_reduce_min_op.py | 49 axis5 = () variable
92 self.axis5 = axis5
114 P.ReduceMin(self.keep_dims5)(self.x5, self.axis5),
|
| D | test_reduce_max_op.py | 49 axis5 = () variable
92 self.axis5 = axis5
114 P.ReduceMax(self.keep_dims5)(self.x5, self.axis5),
|
| D | test_reduce_mean_op.py | 47 axis5 = 3 variable
113 self.axis5 = axis5
159 P.ReduceMean(self.keep_dims5)(self.x5, self.axis5),
209 expect5 = np.mean(x5, axis=axis5, keepdims=keep_dims5)
|
| D | test_reduce_sum_op.py | 48 axis5 = () variable
115 self.axis5 = axis5
161 P.ReduceSum(self.keep_dims5)(self.x5, self.axis5),
|