Searched refs:reference_preprocessor (Results 1 – 2 of 2) sorted by relevance
236 const std::function<void(HloModule*)>& reference_preprocessor) { in MakeReferenceModule() argument240 if (reference_preprocessor != nullptr) { in MakeReferenceModule()241 reference_preprocessor(reference_module.get()); in MakeReferenceModule()256 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompareInternal() argument259 MakeReferenceModule(*module, reference_preprocessor)); in RunAndCompareInternal()276 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompare() argument279 /*run_hlo_passes=*/true, reference_preprocessor); in RunAndCompare()290 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompareNoHloPasses() argument293 /*run_hlo_passes=*/false, reference_preprocessor); in RunAndCompareNoHloPasses()302 const std::function<void(HloModule*)>& reference_preprocessor) { in RunAndCompare() argument[all …]
206 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)215 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)221 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)228 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)237 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)249 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)254 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)258 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)326 const std::function<void(HloModule*)>& reference_preprocessor);335 const std::function<void(HloModule*)>& reference_preprocessor);