Home
last modified time | relevance | path

Searched defs:tracked_device_buffer (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.cc471 auto tracked_device_buffer = std::make_unique<TrackedTfrtCpuDeviceBuffer>( in CreateViewOfDeviceBuffer() local
579 auto tracked_device_buffer = std::make_unique<TrackedTfrtCpuDeviceBuffer>( in BufferFromHostBuffer() local
650 std::unique_ptr<TrackedTfrtCpuDeviceBuffer> tracked_device_buffer, in TfrtCpuBuffer()
701 std::unique_ptr<TrackedTfrtCpuDeviceBuffer> tracked_device_buffer) in TrackedCpuDeviceBufferExternalReference()
1590 auto tracked_device_buffer = std::make_unique<TrackedTfrtCpuDeviceBuffer>( in ExecuteHelper() local
H A Dpjrt_stream_executor_client.cc659 std::shared_ptr<TrackedDeviceBuffer> tracked_device_buffer) in TrackedDeviceBufferExternalReference()