Searched refs:ShouldFuse (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 72 bool CpuInstructionFusion::ShouldFuse(HloInstruction* consumer, in ShouldFuse() function in xla::cpu::CpuInstructionFusion
102 if (!InstructionFusion::ShouldFuse(consumer, operand_index)) { in ShouldFuse()
|
| D | cpu_instruction_fusion.h | 32 bool ShouldFuse(HloInstruction* consumer, int64 operand_index) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion.cc | 251 !InstructionFusion::ShouldFuse(consumer, operand_index)) { in ShouldFuseInexpensiveChecks()
257 bool GpuInstructionFusion::ShouldFuse(HloInstruction* consumer, in ShouldFuse() function in xla::gpu::GpuInstructionFusion
|
| D | instruction_fusion.h | 45 bool ShouldFuse(HloInstruction* consumer, int64 operand_index) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | instruction_fusion.cc | 218 if (do_not_fuse.count(consumer_operand) > 0 || !ShouldFuse(consumer, i)) { in CanFuseOnAllPaths()
498 ShouldFuse(instruction, i)) { in Run()
610 bool InstructionFusion::ShouldFuse(HloInstruction* consumer, in ShouldFuse() function in xla::InstructionFusion
|
| D | instruction_fusion.h | 71 virtual bool ShouldFuse(HloInstruction* consumer, int64 operand_index);
|