Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmulti_output_fusion.cc47 bool GpuMultiOutputFusion::IsFusible(HloInstruction* instr) { in IsFusible() function in xla::gpu::GpuMultiOutputFusion
52 return instr->IsFusible() && in IsFusible()
148 if (!producer->IsFusible()) { in DoProducerConsumerMultiOutputFusion()
Dgpu_fusible.h30 bool IsFusible(const HloInstruction& instr);
Dmulti_output_fusion.h37 bool IsFusible(HloInstruction* instr) override;
Dinstruction_fusion.cc250 if (!IsFusible(*producer) || !IsFusible(*consumer) || in ShouldFuseInexpensiveChecks()
Dgpu_fusible.cc183 bool IsFusible(const HloInstruction& instr) { in IsFusible() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.cc203 if (!consumer->IsFusible()) { in CanFuseOnAllPaths()
306 if (producer->IsFusible() && in ComputeGloballyUnfusible()
480 if (!instruction->IsFusible() && in Run()
490 if (!operand->IsFusible()) { in Run()
Dmulti_output_fusion.cc48 if (!IsFusible(instruction)) { in Run()
66 if (user == instruction || !IsFusible(user)) { in Run()
Dmulti_output_fusion.h69 virtual bool IsFusible(HloInstruction* instr) = 0;
Dhlo_instruction.h1135 bool IsFusible() const;
Dhlo_instructions.cc1382 CHECK(instruction_to_fuse->IsFusible()) << instruction_to_fuse->ToString(); in CloneAndFuseInternal()
Dhlo_instruction.cc2452 bool HloInstruction::IsFusible() const { in IsFusible() function in xla::HloInstruction