Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h247 Status HandleBroadcast(HloInstruction* broadcast) override;
Dhlo_cost_analysis.h113 Status HandleBroadcast(const HloInstruction* broadcast) override;
Ddfs_hlo_visitor_with_default.h195 Status HandleBroadcast(HloInstructionPtr broadcast) override { in HandleBroadcast() function
Dhlo_verifier.h79 Status HandleBroadcast(HloInstruction* broadcast) override;
Ddfs_hlo_visitor.h246 virtual Status HandleBroadcast(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc111 Status HandleBroadcast(HloInstruction* hlo) override;
235 Status DynamicDimensionInferenceVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast() function in xla::DynamicDimensionInferenceVisitor
Dhlo_verifier.cc696 Status ShapeVerifier::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::ShapeVerifier
1801 Status HandleBroadcast(HloInstruction* broadcast) override { in HandleBroadcast() function in xla::__anon45533f2c0811::InstructionVerifier
Dhlo_cost_analysis.cc480 Status HloCostAnalysis::HandleBroadcast(const HloInstruction*) { in HandleBroadcast() function in xla::HloCostAnalysis
Dhlo_evaluator.cc1813 Status HloEvaluator::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::HloEvaluator
Dalgebraic_simplifier.cc289 Status HandleBroadcast(HloInstruction* broadcast) override;
3045 Status AlgebraicSimplifierVisitor::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc3199 return visitor->HandleBroadcast(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h410 Status HandleBroadcast(HloInstruction* hlo) override;
Dspmd_partitioner.cc2237 Status SpmdPartitioningVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast() function in xla::spmd::SpmdPartitioningVisitor