Searched refs:keep_dims0 (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reduce_max_op.py | 29 keep_dims0 = True variable
73 self.keep_dims0 = keep_dims0
109 return (P.ReduceMax(self.keep_dims0)(self.x0, self.axis0),
128 expect0 = np.max(x0, axis=axis0, keepdims=keep_dims0)
222 self.keep_dims0 = keep_dims0
225 return P.ReduceMax(self.keep_dims0)(self.x0, self.axis0)
234 expect = np.max(x0, axis=axis0, keepdims=keep_dims0).astype(np.float64)
243 expect = np.max(x0, axis=axis0, keepdims=keep_dims0).astype(np.float64)
|
| D | test_reduce_all_op.py | 28 keep_dims0 = True variable
49 self.keep_dims0 = keep_dims0
66 return (P.ReduceAll(self.keep_dims0)(self.x0, self.axis0),
80 expect0 = np.all(x0, axis=axis0, keepdims=keep_dims0)
|
| D | test_reduce_any_op.py | 28 keep_dims0 = True variable
49 self.keep_dims0 = keep_dims0
66 return (P.ReduceAny(self.keep_dims0)(self.x0, self.axis0),
80 expect0 = np.any(x0, axis=axis0, keepdims=keep_dims0)
|
| D | test_reduce_sum_op.py | 28 keep_dims0 = True variable
96 self.keep_dims0 = keep_dims0
156 return (P.ReduceSum(self.keep_dims0)(self.x0, self.axis0),
181 expect0 = np.sum(x0, axis=axis0, keepdims=keep_dims0)
314 self.keep_dims0 = keep_dims0
317 return P.ReduceSum(self.keep_dims0)(self.x0, self.axis0)
326 expect = np.sum(x0, axis=axis0, keepdims=keep_dims0).astype(np.float64)
335 expect = np.sum(x0, axis=axis0, keepdims=keep_dims0).astype(np.float64)
|
| D | test_reduce_min_op.py | 29 keep_dims0 = True variable
73 self.keep_dims0 = keep_dims0
109 return (P.ReduceMin(self.keep_dims0)(self.x0, self.axis0),
128 expect0 = np.min(x0, axis=axis0, keepdims=keep_dims0)
|
| D | test_reduce_prod_op.py | 27 keep_dims0 = True variable
71 self.keep_dims0 = keep_dims0
107 return (P.ReduceProd(self.keep_dims0)(self.x0, self.axis0),
|
| D | test_reduce_mean_op.py | 28 keep_dims0 = True variable
94 self.keep_dims0 = keep_dims0
154 return (P.ReduceMean(self.keep_dims0)(self.x0, self.axis0),
179 expect0 = np.mean(x0, axis=axis0, keepdims=keep_dims0)
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_reduce_max.py | 43 keep_dims0 = True
44 expect = get_output(x0, axis0, keep_dims0, False)
45 output = get_output(x0, axis0, keep_dims0, True)
|
| D | test_reduce_min.py | 43 keep_dims0 = True
44 expect = get_output(x0, axis0, keep_dims0, False)
45 output = get_output(x0, axis0, keep_dims0, True)
|