Searched refs:device_handles (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 283 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/ |
| D | service.cc | 347 Backend* backend, absl::Span<const DeviceHandle> device_handles, in ExecuteParallelAndRegisterResult() argument
365 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()
528 if (execution_options.device_handles().empty()) { in GetExecutors()
540 for (const auto& device_handle : execution_options.device_handles()) { in GetExecutors()
559 Replicas(*execute_backend_, execution_options.device_handles(0))); in GetArguments()
575 std::vector<DeviceHandle> device_handles; in ExecuteGraphParallel() local
628 device_handles.insert(device_handles.end(), in ExecuteGraphParallel()
629 execution_options.device_handles().begin(), in ExecuteGraphParallel()
630 execution_options.device_handles().end()); in ExecuteGraphParallel()
[all …]
|
| D | service.h | 265 Backend* backend, absl::Span<const DeviceHandle> device_handles,
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 191 const std::vector<GpuDeviceHandle>& device_handles);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 973 const std::vector<GpuDeviceHandle>& device_handles) { in MapMemory() argument
989 std::vector<CUmemAccessDesc> access_descriptors(device_handles.size()); in MapMemory()
991 access_descriptors[i].location.id = device_handles[i]; in MapMemory()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 352 repeated DeviceHandle device_handles = 5; field
391 repeated DeviceHandle device_handles = 1; field
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 2155 std::vector<xla::int64> device_handles; in TEST() local
2165 device_handles.push_back(outputs[0].scalar<int64>()()); in TEST()
2171 for (size_t i = 0; i + 1 < device_handles.size(); ++i) { in TEST()
2174 {Input(device_handles[i]), Input(device_handles[i + 1])}); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 545 const std::vector<ANeuralNetworksDevice*>& device_handles, in GetTargetSdkVersion() argument
549 for (const auto* device_handle : device_handles) { in GetTargetSdkVersion()
|