Searched defs:HandleGather (Results 1 – 7 of 7) sorted by relevance
256 Status HandleGather(HloInstructionPtr gather) override { in HandleGather() function
1064 Status HloCostAnalysis::HandleGather(const HloInstruction* gather) { in HandleGather() function in xla::HloCostAnalysis
1364 Status DynamicDimensionInferenceVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::DynamicDimensionInferenceVisitor
1654 Status ShapeVerifier::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::ShapeVerifier
2463 Status HloEvaluator::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::HloEvaluator
2786 Status AlgebraicSimplifierVisitor::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::AlgebraicSimplifierVisitor
682 Status SpmdPartitioningVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::spmd::SpmdPartitioningVisitor