Searched refs:axis4 (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_cumsum_op.py | 31 axis4 = 4 variable
43 list2 = [axis0, axis1, axis2, axis3, axis4, axis5, axis6]
59 self.axis4 = axis4
74 self.axis11 = axis4
89 self.axis18 = axis4
104 self.axis25 = axis4
119 self.axis32 = axis4
133 self.cumsum_op(self.x4, self.axis4),
|
| D | test_reduce_op.py | 34 self.axis4 = (0, 1, 2)
48 self.reduce_mean(indice, self.axis4),
59 self.reduce_min(indice, self.axis4),
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_cumsum_op.py | 40 axis4 = 1
65 self.axis4 = axis4
79 P.CumSum()(self.x4, self.axis4),
111 expect4 = np.cumsum(x4, axis=axis4)
|
| D | test_cumprod_op.py | 40 axis4 = 1
65 self.axis4 = axis4
79 P.CumProd()(self.x4, self.axis4),
111 expect4 = np.cumprod(x4, axis=axis4)
|
| D | test_reduce_prod_op.py | 42 axis4 = () variable
86 self.axis4 = axis4
111 P.ReduceProd(self.keep_dims4)(self.x4, self.axis4),
|
| D | test_reduce_min_op.py | 44 axis4 = () variable
88 self.axis4 = axis4
113 P.ReduceMin(self.keep_dims4)(self.x4, self.axis4),
|
| D | test_reduce_max_op.py | 44 axis4 = () variable
88 self.axis4 = axis4
113 P.ReduceMax(self.keep_dims4)(self.x4, self.axis4),
|
| D | test_reduce_mean_op.py | 43 axis4 = 3 variable
109 self.axis4 = axis4
158 P.ReduceMean(self.keep_dims4)(self.x4, self.axis4),
203 expect4 = np.mean(x4, axis=axis4, keepdims=keep_dims4)
|
| D | test_reduce_sum_op.py | 43 axis4 = () variable
111 self.axis4 = axis4
160 P.ReduceSum(self.keep_dims4)(self.x4, self.axis4),
|