Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtracked_tfrt_cpu_device_buffer_test.cc44 TrackedTfrtCpuDeviceBuffer tracked_buffer(/*is_tuple=*/false, {buffer}, in TEST() local
81 TrackedTfrtCpuDeviceBuffer tracked_buffer( in TEST() local
113 TrackedTfrtCpuDeviceBuffer tracked_buffer(/*is_tuple=*/false, {buffer}, in TEST() local
147 TrackedTfrtCpuDeviceBuffer tracked_buffer( in TEST() local
H A Dtfrt_cpu_pjrt_client.cc1376 TrackedTfrtCpuDeviceBuffer* tracked_buffer = nullptr; in ExecuteHelper() local
1418 for (const auto& tracked_buffer : tracked_buffers) { in ExecuteHelper() local