Searched refs:RecordMemoryStats (Results 1 – 3 of 3) sorted by relevance
140 void RecordMemoryStats(const Node* node, const MemoryStats& memory_stats);
320 void CostModel::RecordMemoryStats(const Node* node, in RecordMemoryStats() function in tensorflow::CostModel
368 cm->RecordMemoryStats(node, stats.memory_stats()); in BuildCostModel()