Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc80 Status Fuse(absl::Span<HloInstruction*> fused_fusion_instrs);
91 absl::Span<HloInstruction*> fused_fusion_instrs,
319 absl::Span<HloInstruction*> fused_fusion_instrs, in CreateFusedComputation() argument
325 for (size_t i = 0; i < fused_fusion_instrs.size(); ++i) { in CreateFusedComputation()
326 auto old_params = fused_fusion_instrs[i]->fused_parameters(); in CreateFusedComputation()
328 auto bound_opnd = fused_fusion_instrs[i]->mutable_operand(j); in CreateFusedComputation()
347 for (size_t i = 0; i < fused_fusion_instrs.size(); ++i) { in CreateFusedComputation()
348 auto old_params = fused_fusion_instrs[i]->fused_parameters(); in CreateFusedComputation()
357 for (size_t i = 0; i < fused_fusion_instrs.size(); ++i) { in CreateFusedComputation()
358 auto def_to_use_order = fused_fusion_instrs[i] in CreateFusedComputation()
[all …]