Searched refs:GetOutputSizeOfFusion (Results 1 – 1 of 1) sorted by relevance
45 size_t GetOutputSizeOfFusion(const HloInstruction& instr) { in GetOutputSizeOfFusion() function250 } else if (GetOutputSizeOfFusion(*a) != GetOutputSizeOfFusion(*b)) { in Initialize()251 return GetOutputSizeOfFusion(*a) < GetOutputSizeOfFusion(*b); in Initialize()277 GetOutputSizeOfFusion(*fusion_instrs_.at(right)); in GetNextSpanOfFusions()288 size_t first_output_size = GetOutputSizeOfFusion(*fusion_instrs_[left]); in GetNextSpanOfFusions()298 GetOutputSizeOfFusion(*fusion_instrs_[right])) { in GetNextSpanOfFusions()381 GetOutputSizeOfFusion(*fused_fusion_instrs[0]); in CreateFusedComputation()449 auto num_outputs = GetOutputSizeOfFusion(*fused_instr); in Fuse()