Searched refs:WithMetadata (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_test.cc | 442 sharding.WithMetadata(SingleMetadata(), /*overwrite=*/false); in TEST_F()
451 sharding.WithMetadata(ListMetadata(), /*overwrite=*/false); in TEST_F()
466 sharding.WithMetadata(ListMetadata(), /*overwrite=*/false); in TEST_F()
494 sharding.WithMetadata(SingleMetadata(), /*overwrite=*/true); in TEST_F()
503 sharding.WithMetadata(ListMetadata(), /*overwrite=*/true); in TEST_F()
520 sharding.WithMetadata(ListMetadata(), /*overwrite=*/true); in TEST_F()
|
| D | hlo_sharding.h | 240 HloSharding WithMetadata(absl::Span<const OpMetadata> metadata,
|
| D | hlo_sharding.cc | 662 HloSharding HloSharding::WithMetadata(absl::Span<const OpMetadata> metadata, in WithMetadata() function in xla::HloSharding
|
| D | sharding_propagation.cc | 1485 instruction->sharding().WithMetadata({metadata}, /*overwrite=*/false); in AssignShardingMetadata()
|
| /external/llvm-project/clang/unittests/Tooling/ |
| D | TransformerTest.cpp | 743 TEST_F(TransformerTest, WithMetadata) { in TEST_F() argument
|