Searched defs:layout_func (Results 1 – 1 of 1) sorted by relevance
73 const std::function<xla::Layout(const xla::Shape&)>& layout_func, in AssignLayout()132 const std::function<xla::Layout(const xla::Shape&)>& layout_func, in GetShapeWithLayout()