Searched refs:stop_gradient_ (Results 1 – 3 of 3) sorted by relevance
254 stop_gradient_(false), in CNode()284 bool stop_gradient() const { return stop_gradient_; } in stop_gradient()285 void set_stop_gradient(bool stop_gradient) { stop_gradient_ = stop_gradient; } in set_stop_gradient()363 bool stop_gradient_; variable
37 stop_gradient_(false), in CNode()
45 stop_gradient_(std::make_shared<PrimEliminater>(prim::kPrimStopGradient)), in SpecialOpEliminater()52 eliminaters_.emplace_back(stop_gradient_); in SpecialOpEliminater()77 …OptimizerCallerPtr insert_gradient_of_, stop_gradient_, hook_backward_, print_shape_type_, get_ref…