Searched defs:ds_shape (Results 1 – 1 of 1) sorted by relevance
5024 Shape ds_shape = ShapeUtil::MakeShape(F32, {2, 20, 200}); in TEST_F() local6315 Shape ds_shape = ShapeUtil::MakeShape(F32, slice_sizes); in TEST_P() local6396 Shape ds_shape = ShapeUtil::MakeShape(F32, slice_sizes); in TEST_P() local