Searched refs:ReducePrecisionInsertion (Results 1 – 7 of 7) sorted by relevance
24 std::vector<HloInstruction*> ReducePrecisionInsertion::instructions_to_modify( in instructions_to_modify()68 StatusOr<bool> ReducePrecisionInsertion::insert_after( in insert_after()103 StatusOr<bool> ReducePrecisionInsertion::insert_on_inputs( in insert_on_inputs()163 StatusOr<bool> ReducePrecisionInsertion::insert_on_outputs( in insert_on_outputs()196 StatusOr<bool> ReducePrecisionInsertion::Run(HloModule* module) { in Run()205 computation_changed = ReducePrecisionInsertion::insert_on_inputs( in Run()212 computation_changed = ReducePrecisionInsertion::insert_on_outputs( in Run()232 ReducePrecisionInsertion::InstructionFilterFunction233 ReducePrecisionInsertion::make_filter_function( in make_filter_function()266 HloReducePrecisionOptions ReducePrecisionInsertion::make_options_proto( in make_options_proto()[all …]
31 class ReducePrecisionInsertion : public HloModulePass {39 explicit ReducePrecisionInsertion( in ReducePrecisionInsertion() function51 explicit ReducePrecisionInsertion( in ReducePrecisionInsertion() function59 ~ReducePrecisionInsertion() override{}; in ~ReducePrecisionInsertion()
40 ReducePrecisionInsertion op_insertion(5, 10, location, filter); in InsertOps()537 auto options_proto = ReducePrecisionInsertion::make_options_proto( in TEST_F()542 ReducePrecisionInsertion::make_filter_function(options_proto); in TEST_F()566 auto options_proto = ReducePrecisionInsertion::make_options_proto( in TEST_F()572 ReducePrecisionInsertion::make_filter_function(options_proto); in TEST_F()
272 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto( in XLA_TEST_F()298 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto( in XLA_TEST_F()322 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto( in XLA_TEST_F()347 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto( in XLA_TEST_F()372 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto( in XLA_TEST_F()
90 ReducePrecisionInsertion::AddPasses( in RunHloOptimization()92 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION); in RunHloOptimization()
188 ReducePrecisionInsertion::AddPasses( in OptimizeHloModule()190 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION); in OptimizeHloModule()380 ReducePrecisionInsertion::AddPasses( in OptimizeHloModule()382 ReducePrecisionInsertion::PassTiming::AFTER_FUSION); in OptimizeHloModule()
256 ReducePrecisionInsertion::AddPasses( in RunHloPassesThroughLayoutAssn()258 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION); in RunHloPassesThroughLayoutAssn()331 ReducePrecisionInsertion::AddPasses( in RunHloPassesThroughLayoutAssn()333 ReducePrecisionInsertion::PassTiming::AFTER_FUSION); in RunHloPassesThroughLayoutAssn()