Searched refs:reference_module (Results 1 – 1 of 1) sorted by relevance
237 std::unique_ptr<HloModule> reference_module = test_module.Clone(); in MakeReferenceModule() local241 reference_preprocessor(reference_module.get()); in MakeReferenceModule()243 GetProgramShapeWithLayout(*reference_module))) { in MakeReferenceModule()248 TF_RETURN_IF_ERROR(hlo_verifier_->Run(reference_module.get()).status()); in MakeReferenceModule()249 return std::move(reference_module); in MakeReferenceModule()258 TF_ASSIGN_OR_RETURN(auto reference_module, in RunAndCompareInternal()266 reference_runner_.Execute(std::move(reference_module), in RunAndCompareInternal()