Home
last modified time | relevance | path

Searched refs:HandleGather (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h225 Status HandleGather(HloInstruction* gather) override;
Dhlo_cost_analysis.h122 Status HandleGather(const HloInstruction* gather) override;
Ddfs_hlo_visitor_with_default.h234 Status HandleGather(HloInstructionPtr gather) override { in HandleGather() function
Dhlo_verifier.h108 Status HandleGather(HloInstruction* gather) override;
Ddfs_hlo_visitor.h263 virtual Status HandleGather(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc819 return HandleGather(hlo); in HandleFusion()
1006 Status HloCostAnalysis::HandleGather(const HloInstruction* gather) { in HandleGather() function in xla::HloCostAnalysis
Ddynamic_dimension_inference.cc147 Status HandleGather(HloInstruction* hlo) override;
1188 Status DynamicDimensionInferenceVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::DynamicDimensionInferenceVisitor
Dhlo_verifier.cc1131 Status ShapeVerifier::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::ShapeVerifier
Dhlo_evaluator.cc1717 Status HloEvaluator::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::HloEvaluator
Dalgebraic_simplifier.cc315 Status HandleGather(HloInstruction* gather) override;
2509 Status AlgebraicSimplifierVisitor::HandleGather(HloInstruction* gather) { in HandleGather() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc3253 return visitor->HandleGather(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h417 Status HandleGather(HloInstruction* hlo) override;
Dgather_scatter_handler.cc671 Status SpmdPartitioningVisitor::HandleGather(HloInstruction* hlo) { in HandleGather() function in xla::spmd::SpmdPartitioningVisitor