Home
last modified time | relevance | path

Searched refs:IsInputFusion (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc118 CHECK(instr.IsInputFusion()) in IsReduceInputFusion()
128 CHECK(instr.IsInputFusion()) in IsReduceInputFusion()
Dir_emitter_unnested.cc1877 if (fusion->IsInputFusion()) { in HandleFusion()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.cc1306 if (user->IsLoopFusion() || user->IsInputFusion()) { in CanShareOperandBufferWithUser()
Dhlo_instruction.h1370 bool IsInputFusion() const;
Dhlo_instruction.cc3002 bool HloInstruction::IsInputFusion() const { in IsInputFusion() function in xla::HloInstruction