Home
last modified time | relevance | path

Searched defs:inner_tuple_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_execute_test.cc331 const Shape inner_tuple_shape = in XLA_TEST_F() local
458 const Shape inner_tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F() local
H A Dtuple_test.cc274 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({data_shape, data_shape}); in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment_test.cc871 Shape inner_tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST_P() local