Searched refs:stop_gradient (Results 1 – 16 of 16) sorted by relevance
28 from mindspore.ops.functional import stop_gradient53 c_s = stop_gradient(c)60 c_s = stop_gradient(c)77 c = stop_gradient(o1)85 c_s = stop_gradient(c)173 x = stop_gradient(x)194 ret = stop_gradient(ret)225 c = stop_gradient(c)247 z2 = stop_gradient(z2)262 return stop_gradient(x)[all …]
37 def stop_gradient(dx): function42 stop = P.InsertGradientOf(stop_gradient)
289 bbox_target_ = F.stop_gradient(bbox_target_with_batchsize)290 bbox_weight_ = F.stop_gradient(bbox_weight_with_batchsize)291 label_ = F.stop_gradient(label_with_batchsize)292 label_weight_ = F.stop_gradient(label_weight_with_batchsize)
291 bbox_targets = F.stop_gradient(bbox_targets)292 rcnn_labels = F.stop_gradient(rcnn_labels)317 rois = F.stop_gradient(rois)334 rcnn_masks = F.stop_gradient(rcnn_masks)
159 identity = F.stop_gradient(c2)
168 def stop_gradient(x): function
19 from mindspore.ops.functional import stop_gradient413 samples = stop_gradient(samples)
284 bool stop_gradient() const { return stop_gradient_; } in stop_gradient() function285 void set_stop_gradient(bool stop_gradient) { stop_gradient_ = stop_gradient; } in set_stop_gradient() argument
682 if (cnode_input != nullptr && cnode_input->stop_gradient()) { in BackPropagate()841 if (cnode->stop_gradient()) { in BackPropagate()868 if (!user->isa<CNode>() || !user->cast<CNodePtr>()->stop_gradient()) { in AllReferencesStopped()883 if (!cnode->stop_gradient()) { in PropagateStopGradient()
312 if (cnode_morph->stop_gradient()) { in MapMorphism()729 if (!cnode->stop_gradient()) { in BroadCastStopFlag()752 if (!user->isa<CNode>() || !user->cast<CNodePtr>()->stop_gradient()) { in AllReferencesStopped()
244 stop_gradient = Primitive("stop_gradient") variable
340 ignore_mask = F.stop_gradient(ignore_mask)
546 ignore_mask = F.stop_gradient(ignore_mask)
1006 variance = self.reduce_mean(self.square_diff(x, F.stop_gradient(mean)), self.axis)