Searched refs:primal_debug_infos (Results 1 – 9 of 9) sorted by relevance
36 void SetPrimalDebugInfo(const std::vector<NodeDebugInfoPtr> &primal_debug_infos) { in SetPrimalDebugInfo() argument37 primal_debug_infos_ = primal_debug_infos; in SetPrimalDebugInfo()51 explicit PrimalDebugInfoGuard(const std::vector<NodeDebugInfoPtr> &primal_debug_infos) { in PrimalDebugInfoGuard() argument52 PrimalDebugInfoManager::GetInstance().SetPrimalDebugInfo(primal_debug_infos); in PrimalDebugInfoGuard()
326 std::vector<NodeDebugInfoPtr> primal_debug_infos() { return primal_debug_infos_; } in primal_debug_infos() function349 set_primal_debug_infos(node->primal_debug_infos()); in CloneCNodeInfo()
456 auto primal_debug_infos = nd->primal_debug_infos(); in DumpCNode() local457 if (!primal_debug_infos.empty()) { in DumpCNode()459 for (auto &primal_debug_info : primal_debug_infos) { in DumpCNode()466 auto primal_debug_infos = nd->primal_debug_infos(); in DumpCNode() local467 if (!primal_debug_infos.empty()) { in DumpCNode()469 for (auto &primal_debug_info : primal_debug_infos) { in DumpCNode()
187 auto primal_debug_infos = cnode->primal_debug_infos(); in GetSourceLineList() local189 for (auto &primal_debug_info : primal_debug_infos) { in GetSourceLineList()
171 const std::vector<NodeDebugInfoPtr> &primal_debug_infos);189 const std::vector<NodeDebugInfoPtr> &primal_debug_infos) { in BpropToK() argument197 PrimalDebugInfoGuard primal_debug_info_guard(primal_debug_infos); in BpropToK()
440 std::vector<NodeDebugInfoPtr> primal_debug_infos; in KPrimitive() local445 primal_debug_infos.push_back(user_iter->first->debug_info()); in KPrimitive()453 auto expanded_fg = BpropToK(prim, bprop_fg, nullptr, cnode, primal_attrs, primal_debug_infos); in KPrimitive()
49 node->set_primal_debug_infos(cnode->primal_debug_infos()); in CloneCNode()
96 cp_node->set_primal_debug_infos(cnode->primal_debug_infos()); in NewRecomputeNode()
436 cnode->set_primal_debug_infos(ori_cnode->primal_debug_infos()); in NewCNodeWithInfos()