Searched refs:ResetLayout (Results 1 – 4 of 4) sorted by relevance
73 void ResetLayout(const Layout& layout);77 void ResetLayout(const Layout& layout, ShapeIndexView shape_index);
63 void ShapeLayout::ResetLayout(const Layout& layout) { in ResetLayout() function in xla::ShapeLayout70 void ShapeLayout::ResetLayout(const Layout& layout, in ResetLayout() function in xla::ShapeLayout
270 ->ResetLayout(layout); in ForceParameterLayout()278 ->ResetLayout(layout); in ForceResultLayout()285 ->ResetLayout(layout, shape_index); in ForceResultLayout()
732 computation_layout.mutable_result_layout()->ResetLayout( in TEST_F()869 computation_layout.mutable_result_layout()->ResetLayout( in TEST_F()913 computation_layout.mutable_result_layout()->ResetLayout( in TEST_F()