Home
last modified time | relevance | path

Searched refs:buffer_pointers (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc170 std::vector<void*> buffer_pointers; in ExecuteComputeFunction() local
172 buffer_pointers.push_back( in ExecuteComputeFunction()
178 buffer_pointers.size()); in ExecuteComputeFunction()
183 absl::StrJoin(buffer_pointers, ", ", ptr_printer)); in ExecuteComputeFunction()
192 compute_function_(nullptr, run_options, nullptr, buffer_pointers.data(), in ExecuteComputeFunction()
H A Dcpu_xfeed.cc138 std::vector<cpu::runtime::XfeedBuffer*> buffer_pointers; in TransferBuffersFromOutfeedInternal() local
139 buffer_pointers.reserve(buffers.size()); in TransferBuffersFromOutfeedInternal()
141 buffer_pointers.push_back(b.get()); in TransferBuffersFromOutfeedInternal()
146 xfeed_manager->outfeed()->EnqueueBuffersAtomically(buffer_pointers); in TransferBuffersFromOutfeedInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.cc1448 std::vector<void*> buffer_pointers; in ExecuteHelper() local
1449 buffer_pointers.reserve(buffer_table.size()); in ExecuteHelper()
1451 buffer_pointers.push_back(buffer->data()); in ExecuteHelper()
1453 void* result_buffer = buffer_pointers[result_buffer_index_]; in ExecuteHelper()
1490 buffer_pointers.data(), &status, in ExecuteHelper()
1519 buffer_pointers = std::move(buffer_pointers), in ExecuteHelper()
1546 nullptr, buffer_pointers.data(), in ExecuteHelper()