Searched refs:WithoutMetadata (Results 1 – 5 of 5) sorted by relevance
535 TEST_F(HloShardingTest, WithoutMetadata) { in TEST_F() argument538 auto sharding_no_metadata = sharding.WithoutMetadata(); in TEST_F()544 auto sharding_no_metadata = sharding.WithoutMetadata(); in TEST_F()556 auto sharding_no_metadata = sharding.WithoutMetadata(); in TEST_F()
234 HloSharding WithoutMetadata() const;
681 HloSharding HloSharding::WithoutMetadata() const { in WithoutMetadata() function in xla::HloSharding
44 instruction->set_sharding(instruction->sharding().WithoutMetadata()); in ClearMetadata()1705 while_is_sharded(module.get(), sharding.WithoutMetadata(), in TEST_P()
1505 instruction->sharding().WithoutMetadata(); in RemoveShardingMetadata()