Searched refs:reference_module (Results 1 – 3 of 3) sorted by relevance
51 std::unique_ptr<HloModule> reference_module = in PrepareReferenceModule() local55 reference_module.get())); in PrepareReferenceModule()57 TF_RETURN_IF_ERROR(Despecializer().Run(reference_module.get()).status()); in PrepareReferenceModule()59 return std::move(reference_module); in PrepareReferenceModule()
105 std::unique_ptr<HloModule> reference_module; in RunAndCompare() local109 reference_module = in RunAndCompare()123 if (reference_module == nullptr) { in RunAndCompare()129 ExecuteOnPlatform(std::move(reference_module), args, reference_platform, in RunAndCompare()
224 std::unique_ptr<HloModule> reference_module = test_module.Clone(); in MakeReferenceModule() local228 reference_preprocessor(reference_module.get()); in MakeReferenceModule()230 GetProgramShapeWithLayout(*reference_module))) { in MakeReferenceModule()235 TF_RETURN_IF_ERROR(hlo_verifier_->Run(reference_module.get()).status()); in MakeReferenceModule()236 return std::move(reference_module); in MakeReferenceModule()245 TF_ASSIGN_OR_RETURN(auto reference_module, in RunAndCompareInternal()253 reference_runner_.Execute(std::move(reference_module), in RunAndCompareInternal()