Searched refs:buffers_array (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_transfer_manager.cc | 112 std::vector<uint32_t*> buffers_array; in TransferBuffersToInfeed() local
115 buffers_array.reserve(buffers.size()); in TransferBuffersToInfeed()
119 buffers_array.push_back(const_cast<uint32_t*>(span.data())); in TransferBuffersToInfeed()
124 manager_, tpu_executor->se_executor(), buffers_array.data(), in TransferBuffersToInfeed()
302 char** buffers_array; in LinearizeToBuffers() local
308 manager_, &c_literal, &buffers_array, &buffers_size, &buffers_array_size, in LinearizeToBuffers()
313 memcpy(buf.mutable_data<uint8_t>().data(), buffers_array[i], in LinearizeToBuffers()
319 buffers_array, buffers_size, buffers_array_size); in LinearizeToBuffers()
|
| D | tpu_executor_c_api.h | 207 XLA_TransferManager* manager, XLA_Literal* c_literal, char*** buffers_array,
209 void TpuTransferManager_FreeBuffers(char** buffers_array, int64_t* buffers_size,
217 uint32_t** buffers_array,
|