Searched refs:reference_preprocessor (Results 1 – 2 of 2) sorted by relevance
293 const std::function<void(HloModule*)>& reference_preprocessor) { in MakeReferenceModule() argument297 if (reference_preprocessor != nullptr) { in MakeReferenceModule()298 reference_preprocessor(reference_module.get()); in MakeReferenceModule()313 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompareInternal() argument316 MakeReferenceModule(*module, reference_preprocessor)); in RunAndCompareInternal()337 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompare() argument340 /*run_hlo_passes=*/true, reference_preprocessor); in RunAndCompare()351 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompareNoHloPasses() argument354 /*run_hlo_passes=*/false, reference_preprocessor); in RunAndCompareNoHloPasses()363 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompare() argument[all …]
219 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);227 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);232 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);238 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);250 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);289 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);292 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);295 const std::function<void(HloModule*)>& reference_preprocessor = nullptr);373 const std::function<void(HloModule*)>& reference_preprocessor);382 const std::function<void(HloModule*)>& reference_preprocessor);