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