Searched refs:keep_dims6 (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_prod_op.py | 53 keep_dims6 = False variable
95 self.keep_dims6 = keep_dims6
113 P.ReduceProd(self.keep_dims6)(self.x6, self.axis6),
156 expect6 = np.prod(x6, axis=axis6, keepdims=keep_dims6)
|
| D | test_reduce_min_op.py | 55 keep_dims6 = False variable
97 self.keep_dims6 = keep_dims6
115 P.ReduceMin(self.keep_dims6)(self.x6, self.axis6),
164 expect6 = np.min(x6, axis=axis6, keepdims=keep_dims6)
|
| D | test_reduce_max_op.py | 55 keep_dims6 = False variable
97 self.keep_dims6 = keep_dims6
115 P.ReduceMax(self.keep_dims6)(self.x6, self.axis6),
164 expect6 = np.max(x6, axis=axis6, keepdims=keep_dims6)
|
| D | test_reduce_sum_op.py | 54 keep_dims6 = False variable
120 self.keep_dims6 = keep_dims6
162 P.ReduceSum(self.keep_dims6)(self.x6, self.axis6),
217 expect6 = np.sum(x6, axis=axis6, keepdims=keep_dims6)
|
| D | test_reduce_mean_op.py | 52 keep_dims6 = False variable
118 self.keep_dims6 = keep_dims6
160 P.ReduceMean(self.keep_dims6)(self.x6, self.axis6),
215 expect6 = np.mean(x6, axis=axis6, keepdims=keep_dims6)
|