Home
last modified time | relevance | path

Searched defs:device_handles (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto352 repeated DeviceHandle device_handles = 5; field
391 repeated DeviceHandle device_handles = 1; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc347 Backend* backend, absl::Span<const DeviceHandle> device_handles, in ExecuteParallelAndRegisterResult()
575 std::vector<DeviceHandle> device_handles; in ExecuteGraphParallel() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc382 std::vector<DeviceHandle> device_handles; in GetDeviceHandles() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc973 const std::vector<GpuDeviceHandle>& device_handles) { in MapMemory()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc545 const std::vector<ANeuralNetworksDevice*>& device_handles, in GetTargetSdkVersion()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc2155 std::vector<xla::int64> device_handles; in TEST() local