Searched refs:IsBroadcastTree (Results 1 – 1 of 1) sorted by relevance
99 bool IsBroadcastTree(HloInstruction* op, HloInstruction* consumer,967 IsBroadcastTree(old_producer, producer, to_transform)); in CanPropagate()1483 bool ConvolutionVisitor::IsBroadcastTree( in IsBroadcastTree() function in xla::__anon5e1d62920111::ConvolutionVisitor1503 if (!IsBroadcastTree(op->mutable_operand(i), consumer, in IsBroadcastTree()1855 IsBroadcastTree(consumer->mutable_operand(i), producer, in Propagate()