Home
last modified time | relevance | path

Searched defs:HandleGather (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddfs_hlo_visitor_with_default.h256 Status HandleGather(HloInstructionPtr gather) override { in HandleGather() function
H A Dhlo_cost_analysis.cc1064 Status HloCostAnalysis::HandleGather(const HloInstruction* gather) { in HandleGather() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc1654 Status ShapeVerifier::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::ShapeVerifier
H A Ddynamic_dimension_inference.cc1364 Status DynamicDimensionInferenceVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::DynamicDimensionInferenceVisitor
H A Dhlo_evaluator.cc2463 Status HloEvaluator::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::HloEvaluator
H A Dalgebraic_simplifier.cc2786 Status AlgebraicSimplifierVisitor::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::AlgebraicSimplifierVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dgather_scatter_handler.cc682 Status SpmdPartitioningVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::spmd::SpmdPartitioningVisitor