HomeSort by relevance Sort by last modified time
    Searched refs:ShouldFuse (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_instruction_fusion.h 32 bool ShouldFuse(HloInstruction* consumer, int64 operand_index) override;
cpu_instruction_fusion.cc 64 bool CpuInstructionFusion::ShouldFuse(HloInstruction* consumer,
97 !InstructionFusion::ShouldFuse(consumer, operand_index)) {
99 << "Not fusing: producer has no operands, or !ShouldFuse(consumer).";
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
instruction_fusion.h 30 bool ShouldFuse(HloInstruction* consumer, int64 operand_index) override;
instruction_fusion.cc 45 bool GpuInstructionFusion::ShouldFuse(HloInstruction* consumer,
66 // bitcast, leave it unfused. See `xla::InstructionFusion::ShouldFuse` for
93 InstructionFusion::ShouldFuse(consumer, operand_index);
  /external/tensorflow/tensorflow/compiler/xla/service/
instruction_fusion.h 30 // code generation. Derived classes define ShouldFuse method to select which
62 virtual bool ShouldFuse(HloInstruction* consumer, int64 operand_index);
instruction_fusion.cc 178 if (!ShouldFuse(consumer, i)) {
199 if (!ShouldFuse(consumer, i)) {
342 // We store the original indices of the operands to pass to ShouldFuse.
362 if (!ShouldFuse(instruction, i)) {
421 bool InstructionFusion::ShouldFuse(HloInstruction* consumer,

Completed in 157 milliseconds