Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h149 Status HandleIota(HloInstructionPtr iota) override { in HandleIota() function
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.cc3308 Status AlgebraicSimplifierVisitor::HandleIota(HloInstruction* instruction) { in HandleIota() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2143 Status SpmdPartitioningVisitor::HandleIota(HloInstruction* hlo) { in HandleIota() function in xla::spmd::SpmdPartitioningVisitor