Home
last modified time | relevance | path

Searched refs:IsFusible (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_fusion.cc91 bool IsFusible(Operation* op) { in IsFusible() function
262 if (!IsFusible(op_list_[node_id]) || in Run()
413 if (!IsFusible(op_list_[node_to]) || !IsFusible(op_list_[node_from])) { in TryToContractEdge()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc206 return instr.IsFusible() && in IsInputFusible()
217 return instr.IsFusible() && in IsLoopFusible()
513 return instr.IsFusible() && in IsFusibleAsMultiOutputFusionRoot()
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.cc232 if (!consumer->IsFusible()) { in CanFuseOnAllPaths()
330 if (producer->IsFusible() && in ComputeGloballyUnfusible()
515 if (!instruction->IsFusible() && in Run()
527 if (!operand->IsFusible()) { in Run()
Dmulti_output_fusion.cc52 if (!IsFusible(instruction)) { in Run()
82 if (user == instruction || !IsFusible(user)) { in Run()
Dmulti_output_fusion.h69 virtual bool IsFusible(HloInstruction* instr) = 0;
Dhlo_instruction.h1376 bool IsFusible() const;
Dhlo_instructions.cc1651 CHECK(instruction_to_fuse->IsFusible()) << instruction_to_fuse->ToString(); in CloneAndFuseInternal()
Dhlo_instruction.cc3018 bool HloInstruction::IsFusible() const { in IsFusible() function in xla::HloInstruction