Searched defs:AllocateScopedShapedBuffer (Results 1 – 2 of 2) sorted by relevance
387 StatusOr<ScopedShapedBuffer> TransferManager::AllocateScopedShapedBuffer( in AllocateScopedShapedBuffer() function in xla::TransferManager
72 Status AllocateScopedShapedBuffer( in AllocateScopedShapedBuffer() function