Searched refs:RecordMaxMemorySize (Results 1 – 3 of 3) sorted by relevance
124 void RecordMaxMemorySize(const Node* node, int output_slot, Bytes bytes,
251 void CostModel::RecordMaxMemorySize(const Node* node, int output_slot, in RecordMaxMemorySize() function in tensorflow::CostModel
355 cm->RecordMaxMemorySize(node, output_slot, in BuildCostModel()