Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto468 repeated DeviceHandle device_handles = 5; field
525 repeated DeviceHandle device_handles = 1; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc376 Backend* backend, absl::Span<const DeviceHandle> device_handles, in ExecuteParallelAndRegisterResult()
606 std::vector<DeviceHandle> device_handles; in ExecuteGraphParallel() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc381 std::vector<DeviceHandle> device_handles; in GetDeviceHandles() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_driver.cc918 const std::vector<GpuDeviceHandle>& device_handles) { in MapMemory()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc2162 std::vector<int64_t> device_handles; in TEST() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc651 const std::vector<ANeuralNetworksDevice*>& device_handles, in GetTargetFeatureLevel()