Searched refs:GetProgramShapeWithLayout (Results 1 – 1 of 1) sorted by relevance
75 ProgramShape GetProgramShapeWithLayout(const HloModule& module) { in GetProgramShapeWithLayout() function250 const auto& program_shape = GetProgramShapeWithLayout(test_module); in MakeReferenceModule()255 GetProgramShapeWithLayout(*reference_module))) { in MakeReferenceModule()