Searched refs:keep_dims2 (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_all_op.py | 36 keep_dims2 = True variable
57 self.keep_dims2 = keep_dims2
68 P.ReduceAll(self.keep_dims2)(self.x2, self.axis2),
88 expect2 = np.all(x2, axis=axis2, keepdims=keep_dims2)
|
| D | test_reduce_any_op.py | 36 keep_dims2 = True variable
57 self.keep_dims2 = keep_dims2
68 P.ReduceAny(self.keep_dims2)(self.x2, self.axis2),
88 expect2 = np.any(x2, axis=axis2, keepdims=keep_dims2)
|
| D | test_reduce_prod_op.py | 35 keep_dims2 = True variable
79 self.keep_dims2 = keep_dims2
109 P.ReduceProd(self.keep_dims2)(self.x2, self.axis2),
132 expect2 = np.prod(x2, axis=axis2, keepdims=keep_dims2)
|
| D | test_reduce_min_op.py | 37 keep_dims2 = True variable
81 self.keep_dims2 = keep_dims2
111 P.ReduceMin(self.keep_dims2)(self.x2, self.axis2),
140 expect2 = np.min(x2, axis=axis2, keepdims=keep_dims2)
|
| D | test_reduce_max_op.py | 37 keep_dims2 = True variable
81 self.keep_dims2 = keep_dims2
111 P.ReduceMax(self.keep_dims2)(self.x2, self.axis2),
140 expect2 = np.max(x2, axis=axis2, keepdims=keep_dims2)
|
| D | test_reduce_sum_op.py | 36 keep_dims2 = True variable
104 self.keep_dims2 = keep_dims2
158 P.ReduceSum(self.keep_dims2)(self.x2, self.axis2),
193 expect2 = np.sum(x2, axis=axis2, keepdims=keep_dims2)
|
| D | test_reduce_mean_op.py | 36 keep_dims2 = True variable
102 self.keep_dims2 = keep_dims2
156 P.ReduceMean(self.keep_dims2)(self.x2, self.axis2),
191 expect2 = np.mean(x2, axis=axis2, keepdims=keep_dims2)
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_reduce_max.py | 57 keep_dims2 = True
58 expect = get_output(x2, axis2, keep_dims2, False)
59 output = get_output(x2, axis2, keep_dims2, True)
|
| D | test_reduce_min.py | 57 keep_dims2 = True
58 expect = get_output(x2, axis2, keep_dims2, False)
59 output = get_output(x2, axis2, keep_dims2, True)
|