Searched refs:IsBroadcastTree (Results 1 – 1 of 1) sorted by relevance
91 bool IsBroadcastTree(HloInstruction* op, HloInstruction* consumer,697 IsBroadcastTree(old_producer, producer, to_transform)); in CanPropagate()1071 bool ConvolutionVisitor::IsBroadcastTree( in IsBroadcastTree() function in xla::__anon6418faad0111::ConvolutionVisitor1091 if (!IsBroadcastTree(op->mutable_operand(i), consumer, in IsBroadcastTree()1318 IsBroadcastTree(consumer->mutable_operand(i), producer, in Propagate()