Home
last modified time | relevance | path

Searched refs:TraceGradFprop (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/core/utils/
Dtrace_info.h225 class TraceGradFprop : public TraceInfo {
227 TraceGradFprop() : TraceInfo(nullptr, "grad_fprop", "▶") {} in TraceGradFprop() function
228 explicit TraceGradFprop(const DebugInfoPtr &info) : TraceInfo(info, "grad_fprop", "▶") {} in TraceGradFprop() function
229 MS_DECLARE_PARENT(TraceGradFprop, TraceInfo);
230 ~TraceGradFprop() override = default;
231 …eInfoPtr clone() override { return std::make_shared<TraceGradFprop>(*shared_from_base<TraceGradFpr… in clone()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Ddfunctor.h227 TraceGuard guard(std::make_shared<TraceGradFprop>(outer_debug_info)); in BpropToK()
Dkprim.cc539 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()
Ddfunctor.cc47 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()