Home
last modified time | relevance | path

Searched refs:fusion_instrs_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc101 : fusion_instrs_(), pos_(0) { in FusionCandidates()
111 std::vector<HloInstruction*> fusion_instrs_; member in xla::gpu::__anon03e08b780111::HorizontalLoopFusionImpl::FusionCandidates
235 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 …]