Searched defs:inner_tuple_shape (Results 1 – 2 of 2) sorted by relevance
328 const Shape inner_tuple_shape = in XLA_TEST_F() local453 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