Searched defs:layout_2 (Results 1 – 2 of 2) sorted by relevance
512 auto layout_2 = Layout::FromString("sharding_specs:y, mesh:CPU|x=2,y=2|*CPU") in TEST_F() local524 auto layout_2 = in TEST_F() local533 auto layout_2 = Layout::FromString("sharding_specs:x, mesh:CPU|x=2,y=2|*CPU") in TEST_F() local
128 auto layout_2 = shape_2.mutable_layout(); in TEST() local