Searched defs:HandleBroadcast (Results 1 – 8 of 8) sorted by relevance
147 Status HandleBroadcast(HloInstruction* hlo) override { in HandleBroadcast() function in xla::__anon52bdec210111::LayoutNormalizationVisitor
208 Status HandleBroadcast(HloInstructionPtr broadcast) override { in HandleBroadcast() function
1066 Status ShapeVerifier::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::ShapeVerifier2372 Status HandleBroadcast(HloInstruction* broadcast) override { in HandleBroadcast() function in xla::__anon411d6f1b0e11::InstructionVerifier
481 Status HloCostAnalysis::HandleBroadcast(const HloInstruction*) { in HandleBroadcast() function in xla::HloCostAnalysis
266 Status DynamicDimensionInferenceVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast() function in xla::DynamicDimensionInferenceVisitor
2966 Status HloEvaluator::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::HloEvaluator
3452 Status AlgebraicSimplifierVisitor::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::AlgebraicSimplifierVisitor
2940 Status SpmdPartitioningVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast() function in xla::spmd::SpmdPartitioningVisitor