Searched refs:inner_tuple_shape (Results 1 – 3 of 3) sorted by relevance
331 const Shape inner_tuple_shape = in XLA_TEST_F() local334 ShapeUtil::MakeTupleShape({inner_tuple_shape, vector_shape}); in XLA_TEST_F()456 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F() local457 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()
836 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST_P() local838 ShapeUtil::MakeTupleShape({shape, shape, inner_tuple_shape}); in TEST_P()862 HloInstruction::CreateGetTupleElement(inner_tuple_shape, p, 2)); in TEST_P()