Home
last modified time | relevance | path

Searched defs:HandlePartitionId (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddfs_hlo_visitor_with_default.h135 Status HandlePartitionId(HloInstructionPtr hlo) override { in HandlePartitionId() function
H A Dhlo_cost_analysis.cc838 Status HloCostAnalysis::HandlePartitionId(const HloInstruction* /*hlo*/) { in HandlePartitionId() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc513 Status ShapeVerifier::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::ShapeVerifier
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc1380 Status IrEmitter::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc4213 Status SpmdPartitioningVisitor::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::spmd::SpmdPartitioningVisitor