Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.h47 bool ShouldFuseIntoMultiOutput(HloInstruction* consumer,
Dinstruction_fusion.cc274 bool GpuInstructionFusion::ShouldFuseIntoMultiOutput(HloInstruction* consumer, in ShouldFuseIntoMultiOutput() function in xla::gpu::GpuInstructionFusion
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.h76 virtual bool ShouldFuseIntoMultiOutput(HloInstruction* consumer, in ShouldFuseIntoMultiOutput() function
Dinstruction_fusion.cc501 } else if (ShouldFuseIntoMultiOutput(instruction, i) && in Run()