Home
last modified time | relevance | path

Searched refs:nested_tuple_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc458 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape( in TEST() local
463 EXPECT_TRUE(ShapeUtil::Equal(nested_tuple_shape, in TEST()
464 ShapeUtil::GetSubshape(nested_tuple_shape, {}))); in TEST()
466 array_shape, ShapeUtil::GetSubshape(nested_tuple_shape, {0}))); in TEST()
469 ShapeUtil::GetSubshape(nested_tuple_shape, {1}))); in TEST()
472 ShapeUtil::GetSubshape(nested_tuple_shape, {2, 0}))); in TEST()
487 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape( in TEST() local
492 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {})); in TEST()
493 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {0})); in TEST()
494 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1})); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_test.cc162 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F() local
178 tuple_sharding.GetAsShapeTree(nested_tuple_shape); in TEST_F()
183 EXPECT_IS_OK(tuple_sharding.Validate(nested_tuple_shape, /*num_devices=*/5)); in TEST_F()
Dmemory_space_assignment_test.cc3372 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({shape, tuple_shape}); in TEST_P() local
3389 HloInstruction::CreateParameter(0, nested_tuple_shape, "p")); in TEST_P()
3404 nested_tuple_shape, HloInstruction::FusionKind::kCustom, {p0, p0}, in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc333 const Shape nested_tuple_shape = in XLA_TEST_F() local
339 auto param = Parameter(&builder, 0, nested_tuple_shape, "param"); in XLA_TEST_F()