Searched refs:BFloat16ConversionFolding (Results 1 – 3 of 3) sorted by relevance
34 class BFloat16ConversionFolding : public HloModulePass {36 explicit BFloat16ConversionFolding(const BFloat16Support* bfloat16_support) in BFloat16ConversionFolding() function39 ~BFloat16ConversionFolding() override = default;
239 StatusOr<bool> BFloat16ConversionFolding::Run(HloModule* module) { in Run()
76 BFloat16ConversionFolding fold(&bfloat16_support_); in FoldConversions()