Home
last modified time | relevance | path

Searched refs:Replicas (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc365 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handles[i])); in ExecuteParallelAndRegisterResult()
374 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handles[i])); in ExecuteParallelAndRegisterResult()
477 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handle)); in ExecuteAndRegisterResult()
542 Replicas(*execute_backend_, device_handle)); in GetExecutors()
559 Replicas(*execute_backend_, execution_options.device_handles(0))); in GetArguments()
839 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*execute_backend_, in Execute()
957 Replicas(*execute_backend_, arg->device_handle())); in TransferToServer()
960 replicas, Replicas(*execute_backend_, SingleComputationDeviceHandle())); in TransferToServer()
1000 Replicas(*execute_backend_, arg->device_handle())); in TransferToInfeed()
1005 Replicas(*execute_backend_, SingleComputationDeviceHandle())); in TransferToInfeed()
[all …]
Dservice.h277 StatusOr<std::vector<se::StreamExecutor*>> Replicas(