Searched defs:layout_b (Results 1 – 3 of 3) sorted by relevance
193 const Layout& layout_a, const Layout& layout_b, in GetBroadcastLayoutForElementWise()
1158 const Layout& layout_b) { in ConcatenateLayouts()
3182 def _to_from_layout(layout_a, layout_b, a): argument