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