Searched defs:inner_tuple_shape (Results 1 – 3 of 3) sorted by relevance
331 const Shape inner_tuple_shape = in XLA_TEST_F() local456 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F() local
434 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({data_shape, data_shape}); in XLA_TEST_F() local
836 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST_P() local