Searched refs:axis8 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_prod_op.py | 60 axis8 = () variable
102 self.axis8 = axis8
115 P.ReduceProd(self.keep_dims8)(self.x8, self.axis8))
|
| D | test_reduce_min_op.py | 62 axis8 = () variable
104 self.axis8 = axis8
117 P.ReduceMin(self.keep_dims8)(self.x8, self.axis8))
|
| D | test_reduce_max_op.py | 62 axis8 = () variable
104 self.axis8 = axis8
117 P.ReduceMax(self.keep_dims8)(self.x8, self.axis8))
|
| D | test_reduce_sum_op.py | 61 axis8 = (1, 2) variable
127 self.axis8 = axis8
164 P.ReduceSum(self.keep_dims8)(self.x8, self.axis8),
229 expect8 = np.sum(x8, axis=axis8, keepdims=keep_dims8)
|
| D | test_reduce_mean_op.py | 59 axis8 = (1, 2) variable
125 self.axis8 = axis8
162 P.ReduceMean(self.keep_dims8)(self.x8, self.axis8),
227 expect8 = np.mean(x8, axis=axis8, keepdims=keep_dims8)
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_cumsum_op.py | 68 self.axis8 = axis1
137 self.cumsum_op(self.x8, self.axis8),
|