Searched refs:TraceGradSens (Results 1 – 2 of 2) sorted by relevance
243 class TraceGradSens : public TraceInfo {245 TraceGradSens() : TraceInfo(nullptr, "grad_sens", "∇") {} in TraceGradSens() function246 explicit TraceGradSens(const DebugInfoPtr &info) : TraceInfo(info, "grad_sens", "∇") {} in TraceGradSens() function247 MS_DECLARE_PARENT(TraceGradSens, TraceInfo);248 ~TraceGradSens() override = default;249 …ceInfoPtr clone() override { return std::make_shared<TraceGradSens>(*shared_from_base<TraceGradSen… in clone()
264 TraceGuard guard(std::make_shared<TraceGradSens>(out_param->debug_info())); in BpropToK()