Searched refs:TrackNode (Results 1 – 2 of 2) sorted by relevance
114 void TrackNode(int64 step, const GraphNode* node);138 void TrackNode(const GraphNode* node) { mem_tracker_.TrackNode(step_, node); } in TrackNode() function
147 void MemoryTracker::TrackNode(int64 step, const GraphNode* node) { in TrackNode() function in tensorflow::tfprof::MemoryTracker176 TrackNode(gnode); in AllocateTimeNodes()