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