Searched defs:HandleGather (Results 1 – 7 of 7) sorted by relevance
234 Status HandleGather(HloInstructionPtr gather) override { in HandleGather() function
1006 Status HloCostAnalysis::HandleGather(const HloInstruction* gather) { in HandleGather() function in xla::HloCostAnalysis
1131 Status ShapeVerifier::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::ShapeVerifier
1188 Status DynamicDimensionInferenceVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::DynamicDimensionInferenceVisitor
1717 Status HloEvaluator::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::HloEvaluator
2509 Status AlgebraicSimplifierVisitor::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::AlgebraicSimplifierVisitor
671 Status SpmdPartitioningVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::spmd::SpmdPartitioningVisitor