Searched refs:TraceGradOperation (Results 1 – 2 of 2) sorted by relevance
266 class TraceGradOperation : public TraceInfo {268 explicit TraceGradOperation(const DebugInfoPtr &info) : TraceInfo(info, "grad_ops", "") {} in TraceGradOperation() function269 MS_DECLARE_PARENT(TraceGradOperation, TraceInfo);270 ~TraceGradOperation() override = default;272 return std::make_shared<TraceGradOperation>(*shared_from_base<TraceGradOperation>()); in clone()
680 TraceGuard g(std::make_shared<TraceGradOperation>(forward_graph->debug_info())); in GenerateFuncGraph()703 TraceGuard guard(std::make_shared<TraceGradOperation>(forward_graph->debug_info())); in GenerateFuncGraph()