Home
last modified time | relevance | path

Searched refs:HandleAllToAll (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h77 Status HandleAllToAll(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h103 Status HandleAllToAll(HloInstructionPtr hlo) override { in HandleAllToAll() function
Dhlo_verifier.h58 Status HandleAllToAll(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h117 virtual Status HandleAllToAll(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc593 Status HloCostAnalysis::HandleAllToAll(const HloInstruction* hlo) { in HandleAllToAll() function in xla::HloCostAnalysis
Dhlo_verifier.cc195 Status ShapeVerifier::HandleAllToAll(HloInstruction* hlo) { in HandleAllToAll() function in xla::ShapeVerifier
Dhlo_instruction.cc2552 return visitor->HandleAllToAll(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h147 Status HandleAllToAll(HloInstruction* instruction) override;
Dir_emitter.cc1900 Status IrEmitter::HandleAllToAll(HloInstruction*) { in HandleAllToAll() function in xla::cpu::IrEmitter