Searched refs:keep_dims7 (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_prod_op.py | 57 keep_dims7 = True variable
99 self.keep_dims7 = keep_dims7
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 | 59 keep_dims7 = True variable
101 self.keep_dims7 = keep_dims7
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 | 59 keep_dims7 = True variable
101 self.keep_dims7 = keep_dims7
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 | 58 keep_dims7 = True variable
124 self.keep_dims7 = keep_dims7
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 | 56 keep_dims7 = True variable
122 self.keep_dims7 = keep_dims7
161 P.ReduceMean(self.keep_dims7)(self.x7, self.axis7),
221 expect7 = np.mean(x7, axis=axis7, keepdims=keep_dims7)
|