D | costmodel.cc | 61 Ensure(global_id, num_slots); in MergeFromLocal() 85 Ensure(i, num_slots); in MergeFromGlobal() 109 Ensure(global_id, ns.output_size()); in MergeFromStats() 125 void CostModel::Ensure(int id, int num_outputs) { in Ensure() function in tensorflow::CostModel 158 Ensure(id, 0); in SetNumOutputs() 164 Ensure(id, num_outputs); in SetNumOutputs() 213 Ensure(id, node->num_outputs()); in RecordTime() 261 Ensure(id, node->num_outputs()); in RecordMaxMemorySize() 337 Ensure(id, node->num_outputs()); in RecordMaxExecutionTime() 353 Ensure(id, node->num_outputs()); in RecordAllocationId()
|