Home
last modified time | relevance | path

Searched refs:IsLoopFusible (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc210 bool IsLoopFusible(const HloInstruction& instr) { in IsLoopFusible() function
243 if (!IsLoopFusible(producer)) { in IsProducerConsumerFusible()
248 if (!IsInputFusible(consumer) && !IsLoopFusible(consumer)) { in IsProducerConsumerFusible()
303 if (!IsLoopFusible(producer) || !IsFusibleAsMultiOutputFusionRoot(consumer)) { in IsProducerConsumerMultiOutputFusible()
Dgpu_fusible.h31 bool IsLoopFusible(const HloInstruction& instr);