Home
last modified time | relevance | path

Searched defs:HandleBroadcast (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_normalization.cc147 Status HandleBroadcast(HloInstruction* hlo) override { in HandleBroadcast() function in xla::__anon314bcc100111::LayoutNormalizationVisitor
Ddfs_hlo_visitor_with_default.h208 Status HandleBroadcast(HloInstructionPtr broadcast) override { in HandleBroadcast() function
Dhlo_verifier.cc1066 Status ShapeVerifier::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::ShapeVerifier
2372 Status HandleBroadcast(HloInstruction* broadcast) override { in HandleBroadcast() function in xla::__anonb51dd00a0e11::InstructionVerifier
Dhlo_cost_analysis.cc481 Status HloCostAnalysis::HandleBroadcast(const HloInstruction*) { in HandleBroadcast() function in xla::HloCostAnalysis
Ddynamic_dimension_inference.cc266 Status DynamicDimensionInferenceVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast() function in xla::DynamicDimensionInferenceVisitor
Dhlo_evaluator.cc2966 Status HloEvaluator::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::HloEvaluator
Dalgebraic_simplifier.cc3452 Status AlgebraicSimplifierVisitor::HandleBroadcast(HloInstruction* broadcast) { in HandleBroadcast() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2940 Status SpmdPartitioningVisitor::HandleBroadcast(HloInstruction* hlo) { in HandleBroadcast() function in xla::spmd::SpmdPartitioningVisitor