Searched refs:HandleGather (Results 1 – 13 of 13) sorted by relevance
225 Status HandleGather(HloInstruction* gather) override;
122 Status HandleGather(const HloInstruction* gather) override;
234 Status HandleGather(HloInstructionPtr gather) override { in HandleGather() function
108 Status HandleGather(HloInstruction* gather) override;
263 virtual Status HandleGather(HloInstructionPtr hlo) = 0;
819 return HandleGather(hlo); in HandleFusion()1006 Status HloCostAnalysis::HandleGather(const HloInstruction* gather) { in HandleGather() function in xla::HloCostAnalysis
147 Status HandleGather(HloInstruction* hlo) override;1188 Status DynamicDimensionInferenceVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::DynamicDimensionInferenceVisitor
1131 Status ShapeVerifier::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::ShapeVerifier
1717 Status HloEvaluator::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::HloEvaluator
315 Status HandleGather(HloInstruction* gather) override;2509 Status AlgebraicSimplifierVisitor::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::AlgebraicSimplifierVisitor
3253 return visitor->HandleGather(this); in Visit()
417 Status HandleGather(HloInstruction* hlo) override;
671 Status SpmdPartitioningVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::spmd::SpmdPartitioningVisitor