Searched refs:AllocateScopedShapedBuffer (Results 1 – 11 of 11) sorted by relevance
269 backend().transfer_manager()->AllocateScopedShapedBuffer( in LiteralToShapedBuffer()319 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferToLocalServer()
107 transfer_manager->AllocateScopedShapedBuffer( in ExecuteOnStream()
146 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAndFetchProfile()153 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAndFetchProfile()
57 ->AllocateScopedShapedBuffer( in AllocateDeviceBuffer()
770 ->AllocateScopedShapedBuffer(start_indices_shape, &allocator, in BM_DynamicSlice()
919 ->AllocateScopedShapedBuffer(shape, &allocator, /*device_ordinal=*/0) in BM_LocalClientOverhead()
222 StatusOr<ScopedShapedBuffer> AllocateScopedShapedBuffer(
100 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferLiteralToDevice()306 backend().transfer_manager()->AllocateScopedShapedBuffer( in ExecuteReplicated()
309 StatusOr<ScopedShapedBuffer> TransferManager::AllocateScopedShapedBuffer( in AllocateScopedShapedBuffer() function in xla::TransferManager
991 execute_backend_->transfer_manager()->AllocateScopedShapedBuffer( in TransferToServer()
85 Status AllocateScopedShapedBuffer( in AllocateScopedShapedBuffer() function190 TF_RETURN_IF_ERROR(AllocateScopedShapedBuffer( in CreateAndTransfer()