D | costmodel.cc | 63 Ensure(global_id, num_slots); in MergeFromLocal() 87 Ensure(i, num_slots); in MergeFromGlobal() 111 Ensure(global_id, ns.output_size()); in MergeFromStats() 127 void CostModel::Ensure(int id, int num_outputs) { in Ensure() function in tensorflow::CostModel 160 Ensure(id, 0); in SetNumOutputs() 166 Ensure(id, num_outputs); in SetNumOutputs() 215 Ensure(id, node->num_outputs()); in RecordTime() 263 Ensure(id, node->num_outputs()); in RecordMaxMemorySize() 339 Ensure(id, node->num_outputs()); in RecordMaxExecutionTime() 355 Ensure(id, node->num_outputs()); in RecordAllocationId()
|