Searched refs:keep_dims3 (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_all_op.py | 40 keep_dims3 = False variable
61 self.keep_dims3 = keep_dims3
69 P.ReduceAll(self.keep_dims3)(self.x3, self.axis3))
92 expect3 = np.all(x3, axis=axis3, keepdims=keep_dims3)
|
| D | test_reduce_any_op.py | 40 keep_dims3 = False variable
61 self.keep_dims3 = keep_dims3
69 P.ReduceAny(self.keep_dims3)(self.x3, self.axis3))
92 expect3 = np.any(x3, axis=axis3, keepdims=keep_dims3)
|
| D | test_reduce_prod_op.py | 39 keep_dims3 = False variable
83 self.keep_dims3 = keep_dims3
110 P.ReduceProd(self.keep_dims3)(self.x3, self.axis3),
138 expect3 = np.prod(x3, axis=axis3, keepdims=keep_dims3)
|
| D | test_reduce_min_op.py | 41 keep_dims3 = False variable
85 self.keep_dims3 = keep_dims3
112 P.ReduceMin(self.keep_dims3)(self.x3, self.axis3),
146 expect3 = np.min(x3, axis=axis3, keepdims=keep_dims3)
|
| D | test_reduce_max_op.py | 41 keep_dims3 = False variable
85 self.keep_dims3 = keep_dims3
112 P.ReduceMax(self.keep_dims3)(self.x3, self.axis3),
146 expect3 = np.max(x3, axis=axis3, keepdims=keep_dims3)
|
| D | test_reduce_sum_op.py | 40 keep_dims3 = False variable
108 self.keep_dims3 = keep_dims3
159 P.ReduceSum(self.keep_dims3)(self.x3, self.axis3),
199 expect3 = np.sum(x3, axis=axis3, keepdims=keep_dims3)
|
| D | test_reduce_mean_op.py | 40 keep_dims3 = False variable
106 self.keep_dims3 = keep_dims3
157 P.ReduceMean(self.keep_dims3)(self.x3, self.axis3),
197 expect3 = np.mean(x3, axis=axis3, keepdims=keep_dims3)
|