Searched refs:src_device_buffer (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 397 std::shared_ptr<TpuSharedBuffer> src_device_buffer = DeviceBuffer(); in CopyToDevice() local
400 on_host_shape_, src_device_buffer, in CopyToDevice()
409 [driver, src_device_buffer](tpu_driver::BufferHandle* dst_handle) { in CopyToDevice()
411 for (auto& event : src_device_buffer->wait_for_use) { in CopyToDevice()
415 src_device_buffer->handle.get(), dst_handle, src_wait_for_use); in CopyToDevice()
422 src_device_buffer->wait_for_use.insert(src_device_buffer->wait_for_use.end(), in CopyToDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1296 std::shared_ptr<TrackedDeviceBuffer> src_device_buffer) { in CopyToDeviceHelper() argument
1305 WaitForBufferDefinitionEventsOnStream(*src_device_buffer, transfer_stream); in CopyToDeviceHelper()
1343 ->ThenRelease(transfer_stream, src_device_buffer); in CopyToDeviceHelper()
1390 ScopedHold src_device_buffer(this, ScopedHold::kUsage); in CopyToDevice() local
1399 AcquireHoldLocked(&src_device_buffer); in CopyToDevice()
1406 src_device_buffer.buffer()); in CopyToDevice()
1421 RecordUsage(std::move(src_device_buffer), in CopyToDevice()
|
| D | pjrt_stream_executor_client.h | 585 std::shared_ptr<TrackedDeviceBuffer> src_device_buffer);
|