Home
last modified time | relevance | path

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

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