Searched refs:inner_tuple_shape (Results 1 – 2 of 2) sorted by relevance
328 const Shape inner_tuple_shape = in XLA_TEST_F() local331 ShapeUtil::MakeTupleShape({inner_tuple_shape, vector_shape}); in XLA_TEST_F()453 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F() local454 std::vector<Shape> inner_tuple_shapes(kFanout, inner_tuple_shape); in XLA_TEST_F()
434 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({data_shape, data_shape}); in XLA_TEST_F() local436 ShapeUtil::MakeTupleShape({inner_tuple_shape, data_shape}); in XLA_TEST_F()