Searched defs:update_layout (Results 1 – 3 of 3) sorted by relevance
33 : update_layout_(update_layout) {} in update_layout_() argument
38 std::function<void(Shape &)> update_layout) { in CreateStartIndicesForCollectiveDecomposition()
63 auto update_layout = [this](xla::Shape* subshape, in AllocateOutputMemoryWithInputReuse() local