Searched refs:WriteTupleIndexTables (Results 1 – 4 of 4) sorted by relevance
118 TF_RETURN_IF_ERROR(WriteTupleIndexTables(stream, device_buffer)); in TransferLiteralToDeviceAsync()
203 Status WriteTupleIndexTables(se::Stream* stream,
233 Status TransferManager::WriteTupleIndexTables( in WriteTupleIndexTables() function in xla::TransferManager
125 transfer_manager->WriteTupleIndexTables(stream.get(), *(buffer->get()))); in AllocateScopedShapedBuffer()477 transfer_manager->WriteTupleIndexTables(stream.get(), tuple_buffers)); in MakeTuple()