Searched refs:GetOutputsTotalSize (Results 1 – 3 of 3) sorted by relevance
78 double GetOutputsTotalSize();
1664 double OperatorInfo::GetOutputsTotalSize() { in GetOutputsTotalSize() function in mindspore::parallel::OperatorInfo
1716 double curr_memory_size = first_op->GetOutputsTotalSize(); in DetermineCriticalOps()1723 curr_memory_size += topo_order[finished]->GetOutputsTotalSize(); in DetermineCriticalOps()1736 curr_memory_size -= prev_op->GetOutputsTotalSize(); in DetermineCriticalOps()