Searched defs:HandleIota (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | dfs_hlo_visitor_with_default.h | 162 Status HandleIota(HloInstructionPtr iota) override { in HandleIota() function |
| D | hlo_cost_analysis.cc | 229 Status HloCostAnalysis::HandleIota(const HloInstruction*) { return OkStatus(); } in HandleIota() function in xla::HloCostAnalysis |
| D | hlo_verifier.cc | 968 Status ShapeVerifier::HandleIota(HloInstruction* hlo) { in HandleIota() function in xla::ShapeVerifier |
| D | algebraic_simplifier.cc | 3691 Status AlgebraicSimplifierVisitor::HandleIota(HloInstruction* instruction) { in HandleIota() function in xla::AlgebraicSimplifierVisitor |
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ | ||
| D | spmd_partitioner.cc | 2800 Status SpmdPartitioningVisitor::HandleIota(HloInstruction* hlo) { in HandleIota() function in xla::spmd::SpmdPartitioningVisitor |