Searched refs:state_tuple_array_shape (Results 1 – 1 of 1) sorted by relevance
31 Shape state_tuple_array_shape = ShapeUtil::MakeShape(xla::S32, {10, 5}); in XLA_TEST_F() local34 ShapeUtil::MakeTupleShape({int_shape, state_tuple_array_shape}); in XLA_TEST_F()