Searched defs:HandlePartitionId (Results 1 – 5 of 5) sorted by relevance
122 Status HandlePartitionId(HloInstructionPtr hlo) override { in HandlePartitionId() function
783 Status HloCostAnalysis::HandlePartitionId(const HloInstruction* /*hlo*/) { in HandlePartitionId() function in xla::HloCostAnalysis
335 Status ShapeVerifier::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::ShapeVerifier
3404 Status SpmdPartitioningVisitor::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::spmd::SpmdPartitioningVisitor
3066 Status IrEmitterUnnested::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::gpu::IrEmitterUnnested