Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc415 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handles[i])); in ExecuteParallelAndRegisterResult()
424 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handles[i])); in ExecuteParallelAndRegisterResult()
525 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handle)); in ExecuteAndRegisterResult()
590 Replicas(*execute_backend_, device_handle)); in GetExecutors()
607 Replicas(*execute_backend_, execution_options.device_handles(0))); in GetArguments()
863 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*execute_backend_, in Execute()
980 Replicas(*execute_backend_, arg->device_handle())); in TransferToServer()
983 replicas, Replicas(*execute_backend_, SingleComputationDeviceHandle())); in TransferToServer()
1023 Replicas(*execute_backend_, arg->device_handle())); in TransferToInfeed()
1028 Replicas(*execute_backend_, SingleComputationDeviceHandle())); in TransferToInfeed()
[all …]
Dservice.h275 StatusOr<std::vector<se::StreamExecutor*>> Replicas(