Searched refs:BFloat16Propagation (Results 1 – 3 of 3) sorted by relevance
35 BFloat16Propagation::BFloat16Propagation( in BFloat16Propagation() function in xla::BFloat16Propagation39 void BFloat16Propagation::DetermineFusionComputationPrecision( in DetermineFusionComputationPrecision()79 void BFloat16Propagation::RevertIfFusionInternalBF16Changes( in RevertIfFusionInternalBF16Changes()165 void BFloat16Propagation::DetermineWhileComputationsPrecision( in DetermineWhileComputationsPrecision()208 void BFloat16Propagation::DetermineConditionalComputationsPrecision( in DetermineConditionalComputationsPrecision()235 bool BFloat16Propagation::AllUsersConsumeBF16(const HloInstruction& hlo, in AllUsersConsumeBF16()369 void BFloat16Propagation::DetermineInstructionPrecision(HloInstruction* hlo, in DetermineInstructionPrecision()451 bool BFloat16Propagation::InstructionIsCandidateForBF16Output( in InstructionIsCandidateForBF16Output()469 void BFloat16Propagation::AdjustCalledComputationParameters( in AdjustCalledComputationParameters()520 void BFloat16Propagation::AdjustCalledComputationRoot(HloInstruction* hlo) { in AdjustCalledComputationRoot()[all …]
63 class BFloat16Propagation : public HloModulePass {65 explicit BFloat16Propagation(const BFloat16Support* bfloat16_support);67 ~BFloat16Propagation() override = default;
68 BFloat16Propagation propagation(&bfloat16_support); in PropagatePrecision()