Searched refs:reshape_candidates (Results 1 – 1 of 1) sorted by relevance
347 HloInstructionSet* reshape_candidates) { in TryReshapeMoveOnCandidates() argument349 while (!reshape_candidates->empty() && removed) { in TryReshapeMoveOnCandidates()351 for (const HloInstruction* instruction : *reshape_candidates) { in TryReshapeMoveOnCandidates()356 for (const HloInstruction* instruction : *reshape_candidates) { in TryReshapeMoveOnCandidates()367 return !reshape_candidates->count(user); in TryReshapeMoveOnCandidates()370 removed |= reshape_candidates->erase(user) > 0; in TryReshapeMoveOnCandidates()376 if (reshape_candidates->empty()) { in TryReshapeMoveOnCandidates()379 for (HloInstruction* instruction : *reshape_candidates) { in TryReshapeMoveOnCandidates()397 HloInstructionSet reshape_candidates; in Run() local400 reshape_candidates.insert(instruction); in Run()[all …]