Home
last modified time | relevance | path

Searched refs:reduce_mean (Results 1 – 25 of 40) sorted by relevance

12

/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dreduce_npu.cc47 auto reduce_mean = new (std::nothrow) hiai::op::ReduceMean(name_); in Init() local
48 if (reduce_mean == nullptr) { in Init()
52 reduce_mean->set_attr_keep_dims(reduce_prim->keep_dims()); in Init()
53 reduce_ = reduce_mean; in Init()
65 auto reduce_mean = reinterpret_cast<hiai::op::ReduceMean *>(reduce_); in SetNPUInputs() local
66 reduce_mean->set_input_x(*npu_inputs[0]).set_input_axes(*npu_inputs[1]); in SetNPUInputs()
/third_party/mindspore/mindspore/nn/probability/zhusuan/variational/
Delbo.py29 self.reduce_mean = P.ReduceMean(keep_dims=False)
45 …elbo = self.reduce_mean(log_prob_x_) + self.reduce_mean(log_prob_z_) - self.reduce_mean(log_prob_z)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_bias_add.py28 self.reduce_mean = P.ReduceMean()
35 loss = self.reduce_mean(loss, (-1,))
74 self.reduce_mean = P.ReduceMean(keep_dims=False).shard(((1, 1, 1, 8),))
79 x = self.reduce_mean(x, -1)
Dtest_reduce_method_info.py319 self.reduce_mean = P.ReduceMean(keep_dims=False).shard(strategy2)
324 out = self.reduce_mean(out, 0)
493 …self.reduce_mean = P.ReduceMean(keep_dims=False).shard(strategy3).add_prim_attr("cross_batch", Tru…
498 out = self.reduce_mean(out, 0)
518 self.reduce_mean = P.ReduceMean(keep_dims=False).shard(strategy2)
523 out = self.reduce_mean(out, -1)
544 self.reduce_mean = P.ReduceMean(keep_dims=False)
549 out = self.reduce_mean(out, -1)
Dtest_auto_parallel_flag.py90 self.reduce_mean = P.ReduceMean()
94 return self.reduce_mean(self.square(diff), get_axis(diff))
Dtest_reshape.py539 self.reduce_mean = P.ReduceMean(keep_dims=reducemean_keep_dims)
543 self.reduce_mean.shard(strategy)
547 x = self.reduce_mean(x, self.reducemean_axis)
556 self.reduce_mean = P.ReduceMean()
563 loss = self.reduce_mean(loss, (-1,))
/third_party/mindspore/tests/st/ops/cpu/
Dtest_reduce_op.py37 self.reduce_mean = P.ReduceMean(False)
44 return (self.reduce_mean(indice, self.axis0),
45 self.reduce_mean(indice, self.axis1),
46 self.reduce_mean(indice, self.axis2),
47 self.reduce_mean(indice, self.axis3),
48 self.reduce_mean(indice, self.axis4),
/third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/
Dtest_ReduceMean.py29 self.reduce_mean = P.ReduceMean(keep_dims=keep_dims)
34 return self.reduce_mean(inputs, self.axis)
/third_party/mindspore/tests/st/fusion/
Dtest_add_relu_buffer_fusion.py33 self.reduce_mean = P.ReduceMean()
40 x = self.reduce_mean(x)
/third_party/mindspore/mindspore/nn/layer/
Dpooling.py216 self.reduce_mean = P.ReduceMean(keep_dims=True)
381 self.reduce_mean = P.ReduceMean(keep_dims=True)
390 x = self.reduce_mean(x, 2)
393 x = self.reduce_mean(x, 2)
Dimage.py271 self.reduce_mean = P.ReduceMean()
287 …nnel, _ = _compute_multi_channel_loss(c1, c2, img1, img2, self.conv, self.concat, self.reduce_mean)
288 loss = self.reduce_mean(ssim_ave_channel, -1)
375 self.reduce_mean = P.ReduceMean()
401 … self.multi_convs_list[i], self.concat, self.reduce_mean)
409 loss = self.reduce_mean(ms_ssim, -1)
Dnormalization.py124 self.reduce_mean = P.ReduceMean(keep_dims=True)
1126 self.reduce_mean = P.ReduceMean(keep_dims=True)
1136 mean = self.reduce_mean(x, 2)
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_reduce_mean.py27 self.reduce_mean = P.ReduceMean(keep_dims=False)
30 return self.reduce_mean(x)
/third_party/mindspore/tests/st/auto_monad/
Dtest_auto_monad_momentum_loss.py57 self.reduce_mean = P.ReduceMean()
61 return self.reduce_mean(self.square(diff), get_axis(diff))
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
D__init__.py46 from .reduce_mean import ReduceMean
/third_party/mindspore/tests/ut/python/optimizer/
Dtest_debug_location.py115 self.reduce_mean = P.ReduceMean()
120 return self.reduce_mean(self.square(diff), get_axis(diff))
Dtest_optimizer_with_loss_scale.py93 self.reduce_mean = P.ReduceMean()
97 return self.reduce_mean(self.square(diff), get_axis(diff))
/third_party/mindspore/mindspore/ops/_op_impl/cpu/
D__init__.py54 from .reduce_mean import _reduce_mean_cpu
/third_party/mindspore/tests/st/gnn/
Daggregator.py213 self.reduce_mean = P.ReduceMean(keep_dims=False)
222 output_feature = self.reduce_mean(input_feature, 1)
/third_party/mindspore/tests/st/networks/models/deeplabv3/src/
Ddeeplabv3.py383 self.reduce_mean = P.ReduceMean()
405 logits = self.reduce_mean(logits, 2)
421 logits = self.reduce_mean(logits, 2)
/third_party/mindspore/tests/st/pynative/loss_scale/
Dtest_loss_scale.py142 self.reduce_mean = P.ReduceMean()
146 return self.reduce_mean(self.square(diff), get_axis(diff))
/third_party/mindspore/tests/st/export_and_load/
Dtest_bgcf.py48 self.reduce_mean = P.ReduceMean(keep_dims=False)
52 neigh_matrix = self.reduce_mean(neigh_feature, 1)
/third_party/mindspore/tests/st/auto_parallel/
Dsoft_entropy_loss_expand_parallel.py153 self.reduce_mean = P.ReduceMean(keep_dims=False).shard(strategy=stra_list[8])
168 loss = self.reduce_mean(loss, -1)
/third_party/mindspore/tests/st/fl/albert/src/
Dcell_wrapper.py148 self.reduce_mean = P.ReduceMean()
159 loss = self.reduce_mean(per_example_loss, self.last_idx)
/third_party/mindspore/tests/ut/python/dtype/
Dtest_list.py224 self.reduce_mean = P.ReduceMean()
232 ret_mean = self.reduce_mean(x, self.axis)

12