Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.h33 HloInstruction::FusionKind ChooseKind(
Dcpu_instruction_fusion.cc171 HloInstruction::FusionKind CpuInstructionFusion::ChooseKind( in ChooseKind() function in xla::cpu::CpuInstructionFusion
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc279 HloInstruction::FusionKind GpuInstructionFusion::ChooseKind( in ChooseKind() function in xla::gpu::GpuInstructionFusion
293 return InstructionFusion::ChooseKind(producer, consumer); in ChooseKind()
Dinstruction_fusion.h50 HloInstruction::FusionKind ChooseKind(
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.h83 virtual HloInstruction::FusionKind ChooseKind(const HloInstruction* producer,
Dinstruction_fusion.cc538 auto kind = ChooseKind(producer, consumer); in AddFusionInstruction()
639 HloInstruction::FusionKind InstructionFusion::ChooseKind( in ChooseKind() function in xla::InstructionFusion