Searched refs:body_computation_layout (Results 1 – 1 of 1) sorted by relevance
865 const ComputationLayout& body_computation_layout) { in CheckWhileLayout() argument870 TF_RET_CHECK(body_computation_layout.parameter_layout(0).MatchesLayoutInShape( in CheckWhileLayout()872 TF_RET_CHECK(body_computation_layout.result_layout().MatchesLayoutInShape( in CheckWhileLayout()