Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/ops/gpu/
Dtest_reduce_mean_op.py269 class ReduceMeanDynamic(nn.Cell): class
271 super(ReduceMeanDynamic, self).__init__()
287 net1 = ReduceMeanDynamic(Tensor(x14), axis14, keepdims=True)
288 net2 = ReduceMeanDynamic(Tensor(x0), axis0, keepdims=True)
309 net = ReduceMeanDynamic(Tensor(x12), axis12, keepdims=False)