Searched refs:sharding_no_metadata (Results 1 – 3 of 3) sorted by relevance
538 auto sharding_no_metadata = sharding.WithoutMetadata(); in TEST_F() local539 EXPECT_TRUE(sharding_no_metadata.metadata().empty()); in TEST_F()544 auto sharding_no_metadata = sharding.WithoutMetadata(); in TEST_F() local545 EXPECT_TRUE(sharding_no_metadata.metadata().empty()); in TEST_F()556 auto sharding_no_metadata = sharding.WithoutMetadata(); in TEST_F() local557 EXPECT_TRUE(sharding_no_metadata.metadata().empty()); in TEST_F()558 ASSERT_TRUE(sharding_no_metadata.IsTuple()); in TEST_F()559 EXPECT_EQ(sharding_no_metadata.tuple_elements().size(), 3); in TEST_F()560 for (const auto& sub_sharding : sharding_no_metadata.tuple_elements()) { in TEST_F()
1504 HloSharding sharding_no_metadata = in RemoveShardingMetadata() local1507 sharding_no_metadata)) { in RemoveShardingMetadata()1508 instruction->set_sharding(std::move(sharding_no_metadata)); in RemoveShardingMetadata()
1899 xla::OpSharding sharding_no_metadata = node_and_sharding->sharding; in FormatNodeAndShardingMsg() local1900 sharding_no_metadata.clear_metadata(); in FormatNodeAndShardingMsg()1902 absl::CEscape(sharding_no_metadata.SerializeAsString()); in FormatNodeAndShardingMsg()