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