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