Searched refs:RecordAllocationId (Results 1 – 3 of 3) sorted by relevance
151 void RecordAllocationId(const Node* node, int output_slot, int64 alloc_id);
349 void CostModel::RecordAllocationId(const Node* node, int output_slot, in RecordAllocationId() function in tensorflow::CostModel
363 cm->RecordAllocationId(node, output_slot, in BuildCostModel()