Searched refs:NodeDebugInfoPtr (Results 1 – 5 of 5) sorted by relevance
36 void SetPrimalDebugInfo(const std::vector<NodeDebugInfoPtr> &primal_debug_infos) { in SetPrimalDebugInfo()40 std::vector<NodeDebugInfoPtr> GetCurrentPrimalDebugInfo() { return primal_debug_infos_; } in GetCurrentPrimalDebugInfo()44 std::vector<NodeDebugInfoPtr> primal_debug_infos_;51 explicit PrimalDebugInfoGuard(const std::vector<NodeDebugInfoPtr> &primal_debug_infos) { in PrimalDebugInfoGuard()
138 NodeDebugInfoPtr debug_info() { in debug_info()145 void set_debug_info(const NodeDebugInfoPtr &debug_info) { in set_debug_info()221 NodeDebugInfoPtr debug_info_;326 std::vector<NodeDebugInfoPtr> primal_debug_infos() { return primal_debug_infos_; } in primal_debug_infos()328 void set_primal_debug_infos(const std::vector<NodeDebugInfoPtr> &debug_infos) { in set_primal_debug_infos()332 void AddPrimalDebugInfo(const NodeDebugInfoPtr debug_info) { in AddPrimalDebugInfo()373 std::vector<NodeDebugInfoPtr> primal_debug_infos_;
171 const std::vector<NodeDebugInfoPtr> &primal_debug_infos);189 const std::vector<NodeDebugInfoPtr> &primal_debug_infos) { in BpropToK()
440 std::vector<NodeDebugInfoPtr> primal_debug_infos; in KPrimitive()
208 using NodeDebugInfoPtr = std::shared_ptr<NodeDebugInfo>; variable