OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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