Home
last modified time | relevance | path

Searched refs:device_handles (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc283 if (!execution_options || execution_options->device_handles().empty()) { in Execute()
291 TF_ASSIGN_OR_RETURN(auto device_handles, in Execute()
293 TF_RET_CHECK(!device_handles.empty()); in Execute()
294 *options_storage->add_device_handles() = std::move(device_handles[0]); in Execute()
382 std::vector<DeviceHandle> device_handles; in GetDeviceHandles() local
383 for (const DeviceHandle& device_handle : response.device_handles()) { in GetDeviceHandles()
384 device_handles.push_back(device_handle); in GetDeviceHandles()
387 return device_handles; in GetDeviceHandles()
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc397 Backend* backend, absl::Span<const DeviceHandle> device_handles, in ExecuteParallelAndRegisterResult() argument
415 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()
576 if (execution_options.device_handles().empty()) { in GetExecutors()
588 for (const auto& device_handle : execution_options.device_handles()) { in GetExecutors()
607 Replicas(*execute_backend_, execution_options.device_handles(0))); in GetArguments()
623 std::vector<DeviceHandle> device_handles; in ExecuteGraphParallel() local
676 device_handles.insert(device_handles.end(), in ExecuteGraphParallel()
677 execution_options.device_handles().begin(), in ExecuteGraphParallel()
678 execution_options.device_handles().end()); in ExecuteGraphParallel()
[all …]
Dservice.h263 Backend* backend, absl::Span<const DeviceHandle> device_handles,
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto297 repeated DeviceHandle device_handles = 5; field
313 repeated DeviceHandle device_handles = 1; field