Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h56 Status HandleIota(const HloInstruction* iota) override;
Ddfs_hlo_visitor_with_default.h149 Status HandleIota(HloInstructionPtr iota) override { in HandleIota() function
Dhlo_verifier.h50 Status HandleIota(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h242 virtual Status HandleIota(HloInstructionPtr hlo) = 0;
Dhlo_evaluator_typed_visitor.h2680 Status HandleIota(HloInstruction* instruction) {
2698 Status HandleIota(HloInstruction* iota) {
2701 Status HandleIota(HloInstruction* iota) override {
2702 return HandleIota<ReturnT>(iota);
Dhlo_cost_analysis.cc224 Status HloCostAnalysis::HandleIota(const HloInstruction*) { in HandleIota() function in xla::HloCostAnalysis
Dhlo_verifier.cc605 Status ShapeVerifier::HandleIota(HloInstruction* hlo) { in HandleIota() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc307 Status HandleIota(HloInstruction* instruction) override;
3308 Status AlgebraicSimplifierVisitor::HandleIota(HloInstruction* instruction) { in HandleIota() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc3263 return visitor->HandleIota(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h438 Status HandleIota(HloInstruction* hlo) override;
Dspmd_partitioner.cc2143 Status SpmdPartitioningVisitor::HandleIota(HloInstruction* hlo) { in HandleIota() function in xla::spmd::SpmdPartitioningVisitor