Searched refs:Sharding (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 27 class Sharding(object): class
45 return Sharding(
57 return Sharding(
85 return Sharding(
117 return Sharding(
170 Sharding.replicate().apply_to_tensor(tensor)
175 Sharding.assign_device(device).apply_to_tensor(tensor)
180 Sharding.tile(tile_assignment).apply_to_tensor(tensor)
185 Sharding.split(tensor, split_dimension, num_devices).apply_to_tensor(tensor)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers_test.cc | 166 ::testing::Not(op::Sharding(HloSharding::AssignDevice(1)))); in TEST()
169 ::testing::Not(op::Sharding(HloSharding::AssignDevice(0)))); in TEST()
170 EXPECT_THAT(p1.get(), op::Sharding(HloSharding::AssignDevice(1))); in TEST()
174 op::Sharding("{{devices=[2]0,1}, {maximal device=1}, {replicated}}")); in TEST()
176 EXPECT_THAT(Explain(p0.get(), op::Sharding(HloSharding::AssignDevice(1))), in TEST()
182 EXPECT_THAT(Explain(p1.get(), op::Sharding(HloSharding::AssignDevice(0))), in TEST()
|
| D | hlo_matchers.h | 372 inline ::testing::Matcher<const ::xla::HloInstruction*> Sharding( in Sharding() function
378 inline ::testing::Matcher<const ::xla::HloInstruction*> Sharding( in Sharding() function
|
| D | hlo.proto | 193 // Sharding for kDomain instructions.
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_configuration.proto | 45 // Sharding strategy of the embedding tables among the hosts.
|