Home
last modified time | relevance | path

Searched refs:keep_dims9 (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/tests/st/ops/gpu/
Dtest_reduce_sum_op.py66 keep_dims9 = False variable
132 self.keep_dims9 = keep_dims9
165 P.ReduceSum(self.keep_dims9)(self.x9, self.axis9),
235 expect9 = np.sum(x9, axis=axis9, keepdims=keep_dims9)
Dtest_reduce_mean_op.py64 keep_dims9 = False variable
130 self.keep_dims9 = keep_dims9
163 P.ReduceMean(self.keep_dims9)(self.x9, self.axis9),
233 expect9 = np.mean(x9, axis=axis9, keepdims=keep_dims9)