Searched refs:WriteTupleIndexTables (Results 1 – 4 of 4) sorted by relevance
113 transfer_manager->WriteTupleIndexTables(stream.get(), *(buffer->get()))); in AllocateScopedShapedBuffer()550 transfer_manager->WriteTupleIndexTables(stream.get(), tuple_buffers)); in MakeTuple()
216 Status WriteTupleIndexTables(se::Stream* stream,
281 Status TransferManager::WriteTupleIndexTables( in WriteTupleIndexTables() function in xla::TransferManager
227 transfer_manager->WriteTupleIndexTables(stream, result.Result())); in AllocateOutputMemoryWithInputReuse()