Searched refs:max_rank_layout (Results 1 – 1 of 1) sorted by relevance
50 const Layout* max_rank_layout; in LayoutsAreReduceInputFusionFriendly() local54 max_rank_layout = ¶m->shape().layout(); in LayoutsAreReduceInputFusionFriendly()59 (LayoutUtil::Equal(param->shape().layout(), *max_rank_layout)); in LayoutsAreReduceInputFusionFriendly()