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