Home
last modified time | relevance | path

Searched refs:sharding_unique_device (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc308 optional<int64> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormTraining()
412 optional<int64> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormInference()
590 auto unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormGrad()
Dbatchnorm_expander_test.cc155 auto device = instruction->sharding_unique_device(); in TEST_F()
Dhlo_domain_test.cc283 auto device = add->sharding_unique_device(); in TEST_F()
425 auto device = assignment.instruction->sharding_unique_device(); in TEST_F()
Dhlo_module_group_metadata.cc288 absl::optional<int64> device = instruction.sharding_unique_device(); in GetInstructionDevice()
Dhlo_instruction.h1150 absl::optional<int64> sharding_unique_device() const { in sharding_unique_device() function