Home
last modified time | relevance | path

Searched refs:HandlePartitionId (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h86 Status HandlePartitionId(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h122 Status HandlePartitionId(HloInstructionPtr hlo) override { in HandlePartitionId() function
Dhlo_verifier.h65 Status HandlePartitionId(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h126 virtual Status HandlePartitionId(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc783 Status HloCostAnalysis::HandlePartitionId(const HloInstruction* /*hlo*/) { in HandlePartitionId() function in xla::HloCostAnalysis
Dhlo_verifier.cc335 Status ShapeVerifier::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::ShapeVerifier
Dhlo_instruction.cc3145 return visitor->HandlePartitionId(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h439 Status HandlePartitionId(HloInstruction* hlo) override;
Dspmd_partitioner.cc3404 Status SpmdPartitioningVisitor::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::spmd::SpmdPartitioningVisitor
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h219 Status HandlePartitionId(HloInstruction* hlo) override;
Dir_emitter_unnested.cc3066 Status IrEmitterUnnested::HandlePartitionId(HloInstruction* hlo) { in HandlePartitionId() function in xla::gpu::IrEmitterUnnested