Searched refs:BFloat16Propagation (Results 1 – 3 of 3) sorted by relevance
33 BFloat16Propagation::BFloat16Propagation( in BFloat16Propagation() function in xla::BFloat16Propagation37 void BFloat16Propagation::DetermineFusionComputationPrecision( in DetermineFusionComputationPrecision()77 void BFloat16Propagation::RevertIfFusionInternalBF16Changes( in RevertIfFusionInternalBF16Changes()163 void BFloat16Propagation::DetermineWhileComputationsPrecision( in DetermineWhileComputationsPrecision()206 bool BFloat16Propagation::AllUsersConsumeBF16(const HloInstruction& hlo, in AllUsersConsumeBF16()311 void BFloat16Propagation::DetermineInstructionPrecision(HloInstruction* hlo, in DetermineInstructionPrecision()388 bool BFloat16Propagation::InstructionIsCandidateForBF16Output( in InstructionIsCandidateForBF16Output()406 void BFloat16Propagation::AdjustCalledComputationParameters( in AdjustCalledComputationParameters()451 void BFloat16Propagation::AdjustCalledComputationRoot(HloInstruction* hlo) { in AdjustCalledComputationRoot()501 bool BFloat16Propagation::ResolveInconsistencyOfAliasingBuffersHelper( in ResolveInconsistencyOfAliasingBuffersHelper()[all …]
63 class BFloat16Propagation : public HloModulePass {65 explicit BFloat16Propagation(const BFloat16Support* bfloat16_support);67 ~BFloat16Propagation() override = default;
68 BFloat16Propagation propagation(&bfloat16_support); in PropagatePrecision()