Searched refs:nested_tuple_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 434 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape( in TEST() local
439 EXPECT_TRUE(ShapeUtil::Equal(nested_tuple_shape, in TEST()
440 ShapeUtil::GetSubshape(nested_tuple_shape, {}))); in TEST()
442 array_shape, ShapeUtil::GetSubshape(nested_tuple_shape, {0}))); in TEST()
445 ShapeUtil::GetSubshape(nested_tuple_shape, {1}))); in TEST()
448 ShapeUtil::GetSubshape(nested_tuple_shape, {2, 0}))); in TEST()
463 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape( in TEST() local
468 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {})); in TEST()
469 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {0})); in TEST()
470 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1})); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_test.cc | 127 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F() local
143 tuple_sharding.GetAsShapeTree(nested_tuple_shape); in TEST_F()
148 EXPECT_IS_OK(tuple_sharding.Validate(nested_tuple_shape, /*num_devices=*/5)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 330 const Shape nested_tuple_shape = in XLA_TEST_F() local
336 auto param = Parameter(&builder, 0, nested_tuple_shape, "param"); in XLA_TEST_F()
|