Searched refs:body_computation_layout (Results 1 – 1 of 1) sorted by relevance
693 const ComputationLayout& body_computation_layout) { in CheckWhileLayout() argument698 TF_RET_CHECK(body_computation_layout.parameter_layout(0).MatchesLayoutInShape( in CheckWhileLayout()701 body_computation_layout.result_layout().MatchesLayoutInShape(init_shape)); in CheckWhileLayout()