Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h137 Status HandlePartitionId(HloInstructionPtr hlo) override { in HandlePartitionId() function
Dhlo_cost_analysis.cc812 Status HloCostAnalysis::HandlePartitionId(const HloInstruction* /*hlo*/) { in HandlePartitionId() function in xla::HloCostAnalysis
Dhlo_verifier.cc497 Status ShapeVerifier::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::ShapeVerifier
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3425 Status SpmdPartitioningVisitor::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::spmd::SpmdPartitioningVisitor