Home
last modified time | relevance | path

Searched defs:src_device_buffer (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc1420 std::shared_ptr<TrackedDeviceBuffer> src_device_buffer) { in CopyToDeviceHelper()
1517 ScopedHold src_device_buffer(this, ScopedHold::kUsage); in CopyToDevice() local
H A Dtfrt_cpu_pjrt_client.cc1044 auto* src_device_buffer = AcquireUsage(usage_event); in CopyToDevice() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc408 std::shared_ptr<TpuSharedBuffer> src_device_buffer = DeviceBuffer(); in CopyToDevice() local