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