Searched refs:TraceGradBprop (Results 1 – 3 of 3) sorted by relevance
234 class TraceGradBprop : public TraceInfo {236 TraceGradBprop() : TraceInfo(nullptr, "grad_bprop", "◀") {} in TraceGradBprop() function237 explicit TraceGradBprop(const DebugInfoPtr &info) : TraceInfo(info, "grad_bprop", "◀") {} in TraceGradBprop() function238 MS_DECLARE_PARENT(TraceGradBprop, TraceInfo);239 ~TraceGradBprop() override = default;240 …eInfoPtr clone() override { return std::make_shared<TraceGradBprop>(*shared_from_base<TraceGradBpr… in clone()
211 cloned_bprop_fg->debug_info()->set_trace_info(std::make_shared<TraceGradBprop>(debug_info)); in BpropToK()
56 TraceGuard guard(std::make_shared<TraceGradBprop>(primal_graph->debug_info())); in DFunctor()