Searched refs:AllocateScopedShapedBuffer (Results 1 – 14 of 14) sorted by relevance
72 Status AllocateScopedShapedBuffer( in AllocateScopedShapedBuffer() function175 TF_RETURN_IF_ERROR(AllocateScopedShapedBuffer(memory_manager, backend, in CreateAndTransfer()202 TF_RETURN_IF_ERROR(AllocateScopedShapedBuffer( in CreateUninitialized()308 AllocateScopedShapedBuffer(memory_manager, backend, device_ordinal(), in SwapIn()
130 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAsyncOnStream()
146 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAndFetchProfile()153 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAndFetchProfile()
101 backend_->transfer_manager()->AllocateScopedShapedBuffer( in RunAndCheck()
57 ->AllocateScopedShapedBuffer( in AllocateDeviceBuffer()
787 ->AllocateScopedShapedBuffer(start_indices_shape, &allocator, in BM_DynamicSlice()
968 ->AllocateScopedShapedBuffer(shape, &allocator, /*device_ordinal=*/0) in BM_LocalClientOverhead()
406 backend().transfer_manager()->AllocateScopedShapedBuffer( in LiteralToShapedBuffer()453 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferToLocalServer()
249 StatusOr<ScopedShapedBuffer> AllocateScopedShapedBuffer(
51 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferLiteralToDevice()240 backend().transfer_manager()->AllocateScopedShapedBuffer( in ExecuteReplicatedImpl()
387 StatusOr<ScopedShapedBuffer> TransferManager::AllocateScopedShapedBuffer( in AllocateScopedShapedBuffer() function in xla::TransferManager
968 execute_backend_->transfer_manager()->AllocateScopedShapedBuffer( in TransferToServer()
324 transfer_manager->AllocateScopedShapedBuffer( in BuildComputationInputs()
376 transfer_manager->AllocateScopedShapedBuffer( in AllocateDestinationBuffer()