Searched refs:fusion_instrs_ (Results 1 – 1 of 1) sorted by relevance
101 : fusion_instrs_(), pos_(0) { in FusionCandidates()111 std::vector<HloInstruction*> fusion_instrs_; member in xla::gpu::__anon81a6c33f0111::HorizontalLoopFusionImpl::FusionCandidates235 fusion_instrs_.push_back(instr); in Initialize()244 fusion_instrs_.begin(), fusion_instrs_.end(), in Initialize()261 if (pos_ >= fusion_instrs_.size()) { in GetNextSpanOfFusions()276 accum_io_size += fusion_instrs_.at(right)->fused_parameters().size() + in GetNextSpanOfFusions()277 GetOutputSizeOfFusion(*fusion_instrs_.at(right)); in GetNextSpanOfFusions()288 size_t first_output_size = GetOutputSizeOfFusion(*fusion_instrs_[left]); in GetNextSpanOfFusions()290 GetUniqueOutputTypeOfFusion(*fusion_instrs_[left]); in GetNextSpanOfFusions()291 for (; right < fusion_instrs_.size(); ++right) { in GetNextSpanOfFusions()[all …]