Searched defs:SetInstructionLayout (Results 1 – 1 of 1) sorted by relevance
354 Status LayoutAssignment::SetInstructionLayout(const Layout& layout, in SetInstructionLayout() function in xla::LayoutAssignment394 Status LayoutAssignment::SetInstructionLayout(const Shape& shape_with_layout, in SetInstructionLayout() function in xla::LayoutAssignment