Searched refs:TraceGradFprop (Results 1 – 4 of 4) sorted by relevance
225 class TraceGradFprop : public TraceInfo {227 TraceGradFprop() : TraceInfo(nullptr, "grad_fprop", "▶") {} in TraceGradFprop() function228 explicit TraceGradFprop(const DebugInfoPtr &info) : TraceInfo(info, "grad_fprop", "▶") {} in TraceGradFprop() function229 MS_DECLARE_PARENT(TraceGradFprop, TraceInfo);230 ~TraceGradFprop() override = default;231 …eInfoPtr clone() override { return std::make_shared<TraceGradFprop>(*shared_from_base<TraceGradFpr… in clone()
227 TraceGuard guard(std::make_shared<TraceGradFprop>(outer_debug_info)); in BpropToK()
539 TraceGuard trace_guard(std::make_shared<TraceGradFprop>(p->debug_info())); in TransformNormalArgs()587 TraceGuard trace_guard(std::make_shared<TraceGradFprop>(p->debug_info())); in TransformArgsForFuncGraph()
47 TraceGuard guard(std::make_shared<TraceGradFprop>(primal_graph->debug_info())); in DFunctor()575 TraceGuard trace_guard(std::make_shared<TraceGradFprop>(primal->debug_info())); in MapParameterToK()