Searched refs:GraphMemory (Results 1 – 2 of 2) sorted by relevance
81 class GraphMemory {83 explicit GraphMemory(uint32_t graph_id) : graph_id_(graph_id), static_mem_size_(0) {} in GraphMemory() function84 ~GraphMemory() = default;111 std::shared_ptr<GraphMemory> AddGraphMemoryNode(uint32_t graph_id);112 std::shared_ptr<GraphMemory> GetGraphMemoryNode(uint32_t graph_id) const;121 std::map<uint32_t, std::shared_ptr<GraphMemory>> graph_memory_;
54 std::shared_ptr<GraphMemory> MemoryProfiling::AddGraphMemoryNode(uint32_t graph_id) { in AddGraphMemoryNode()55 std::shared_ptr<GraphMemory> node = std::make_shared<GraphMemory>(graph_id); in AddGraphMemoryNode()61 std::shared_ptr<GraphMemory> MemoryProfiling::GetGraphMemoryNode(uint32_t graph_id) const { in GetGraphMemoryNode()