| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 75 auto transfer_manager = backend->transfer_manager(); in AllocateScopedShapedBuffer() local
87 xla::Shape on_device_shape = transfer_manager->HostShapeToDeviceShape(shape); in AllocateScopedShapedBuffer()
100 uint64 size = transfer_manager->GetByteSizeRequirement(subshape); in AllocateScopedShapedBuffer()
112 transfer_manager->WriteTupleIndexTables(stream.get(), *(buffer->get()))); in AllocateScopedShapedBuffer()
173 auto transfer_manager = backend->transfer_manager(); in CreateAndTransfer() local
179 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in CreateAndTransfer()
251 auto transfer_manager = backend->transfer_manager(); in StoreToLiteral() local
254 return transfer_manager->TransferLiteralFromDevice(stream.get(), in StoreToLiteral()
273 auto transfer_manager = backend->transfer_manager(); in WriteLiteral() local
275 return transfer_manager->TransferLiteralToDevice(stream.get(), literal, in WriteLiteral()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer_test.cc | 34 client->backend().transfer_manager()->HostShapeToDeviceShape(shape), in MakeArray()
40 client->backend().transfer_manager()->GetByteSizeRequirement( in MakeArray()
68 client->backend().transfer_manager()->HostShapeToDeviceShape(a_shape)); in TEST()
70 client->backend().transfer_manager()->HostShapeToDeviceShape(b_shape)); in TEST()
72 client->backend().transfer_manager()->HostShapeToDeviceShape(c_shape)); in TEST()
110 client->backend().transfer_manager()->HostShapeToDeviceShape( in TEST()
|
| D | pjrt_stream_executor_client.cc | 373 TransferManager* transfer_manager = in AllocateDestinationBuffer() local
374 se_client->client()->backend().transfer_manager(); in AllocateDestinationBuffer()
376 transfer_manager->AllocateScopedShapedBuffer( in AllocateDestinationBuffer()
387 DCHECK(transfer_manager->CanShapedBufferBeAccessedNow( in AllocateDestinationBuffer()
430 DCHECK(transfer_manager->CanShapedBufferBeAccessedNow( in AllocateDestinationBuffer()
435 TF_RETURN_IF_ERROR(transfer_manager->WriteTupleIndexTablesAsync( in AllocateDestinationBuffer()
663 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostBuffer() local
665 transfer_manager->ChooseCompactLayoutForShape(shape)); in BufferFromHostBuffer()
762 auto transfer_h2d = [local_client = client(), transfer_manager, local_device, in BufferFromHostBuffer()
791 TF_CHECK_OK(transfer_manager->TransferLiteralToDeviceAsync( in BufferFromHostBuffer()
[all …]
|
| /external/crosvm/devices/src/usb/xhci/ |
| D | transfer_ring_controller.rs | 31 transfer_manager: XhciTransferManager, field
40 let xhci_transfer = self.transfer_manager.create_transfer( in handle_transfer_descriptor()
57 self.transfer_manager.cancel_all(); in stop()
84 transfer_manager: XhciTransferManager::new(), in new()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 279 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in BuildComputationInputs() local
282 transfer_manager->HostShapeToDeviceShape(input_host_shape)); in BuildComputationInputs()
286 transfer_manager->GetByteSizeRequirement(input_buffers->buffers.shape()); in BuildComputationInputs()
324 transfer_manager->AllocateScopedShapedBuffer( in BuildComputationInputs()
412 xla::TransferManager* const transfer_manager = in AllocateOutputTensors() local
413 node_context->backend()->transfer_manager(); in AllocateOutputTensors()
445 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in AllocateOutputTensors()
657 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoWork() local
692 if (transfer_manager->CanShapedBufferBeAccessedNow(stream->parent(), in DoWork()
694 TF_RETURN_IF_ERROR(transfer_manager->WriteRootTupleIndexTable( in DoWork()
[all …]
|
| D | outfeed_ops.cc | 43 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork() argument
57 transfer_manager->TransferLiteralFromOutfeed(stream_executor, literal)); in DoWork()
87 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork() argument
100 transfer_manager->TransferLiteralFromOutfeed(stream_executor, literal)); in DoWork()
|
| D | infeed_ops.h | 35 xla::TpuTransferManagerInterface* transfer_manager,
54 xla::TpuTransferManagerInterface* transfer_manager,
|
| D | outfeed_ops.h | 34 xla::TpuTransferManagerInterface* transfer_manager,
55 xla::TpuTransferManagerInterface* transfer_manager,
|
| D | infeed_ops.cc | 385 xla::TpuTransferManagerInterface* transfer_manager, in DoWork() argument
391 TF_RETURN_IF_ERROR(transfer_manager->TransferBuffersToInfeed( in DoWork()
418 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork() argument
446 transfer_manager->TransferLiteralToInfeed(stream_executor, literal)); in DoWork()
472 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork() argument
514 transfer_manager->TransferLiteralToInfeed(stream_executor, tuple)); in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.cc | 105 TF_ASSIGN_OR_RETURN(TransferManager * transfer_manager, in ExecuteAsyncOnStream()
113 transfer_manager->TransferLiteralFromDevice( in ExecuteAsyncOnStream()
130 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAsyncOnStream()
133 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in ExecuteAsyncOnStream()
|
| D | BUILD | 18 "//tensorflow/compiler/xla/service:transfer_manager",
93 "//tensorflow/compiler/xla/service:transfer_manager",
122 "//tensorflow/compiler/xla/service:transfer_manager",
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 207 backend->transfer_manager()->TransferLiteralFromDevice( in DumpArguments()
226 backend->transfer_manager()->TransferLiteralFromDevice( in DumpOutputsAndSaveSnapshot()
406 backend().transfer_manager()->AllocateScopedShapedBuffer( in LiteralToShapedBuffer()
410 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in LiteralToShapedBuffer()
419 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in ShapedBufferToLiteral()
432 return backend().transfer_manager()->TransferLiteralToInfeed(executor, in TransferToInfeedLocal()
440 return backend().transfer_manager()->TransferLiteralFromOutfeed(executor, in TransferFromOutfeedLocal()
453 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferToLocalServer()
457 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferToLocalServer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 89 TF_ASSIGN_OR_RETURN(auto transfer_manager, in CreateBackend()
94 new Backend(platform, compiler, stream_executors, transfer_manager, in CreateBackend()
123 TransferManager* transfer_manager, in Backend() argument
128 transfer_manager_(transfer_manager), in Backend()
|
| D | hlo_runner.cc | 51 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferLiteralToDevice()
56 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferLiteralToDevice()
87 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in TransferLiteralFromDevice()
240 backend().transfer_manager()->AllocateScopedShapedBuffer( in ExecuteReplicatedImpl()
242 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in ExecuteReplicatedImpl()
270 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralToInfeed( in ExecuteReplicatedImpl()
289 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in ExecuteReplicatedImpl()
312 backend().transfer_manager()->TransferLiteralFromDevice( in ExecuteReplicatedImpl()
|
| D | service.cc | 75 se::Stream* stream, TransferManager* transfer_manager, in RecordArguments() argument
81 transfer_manager->TransferLiteralFromDevice(stream, *argument)); in RecordArguments()
89 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult() argument
93 transfer_manager->TransferLiteralFromDevice(stream, result)); in RecordResult()
656 execute_backend_->transfer_manager(), in ExecuteGraphParallel()
722 execute_backend_->transfer_manager(), in ExecuteGraphParallel()
876 execute_backend_->transfer_manager(), &snapshot)); in Execute()
892 execute_backend_->transfer_manager(), in Execute()
936 execute_backend_->transfer_manager()->TransferLiteralFromDevice( in TransferToClient()
968 execute_backend_->transfer_manager()->AllocateScopedShapedBuffer( in TransferToServer()
[all …]
|
| D | backend.h | 94 TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
161 TransferManager* transfer_manager,
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 36 TF_ASSIGN_OR_RETURN(auto transfer_manager, TransferManager::GetForPlatform( in PopulateResultTupleBuffers()
38 if (transfer_manager->CanShapedBufferBeAccessedNow(stream->parent(), in PopulateResultTupleBuffers()
40 TF_RETURN_IF_ERROR(transfer_manager->WriteTupleIndexTablesAsync( in PopulateResultTupleBuffers()
47 return transfer_manager->WriteTupleIndexTablesAsync(stream, result); in PopulateResultTupleBuffers()
|
| D | BUILD | 217 "//tensorflow/compiler/xla/service:transfer_manager",
236 "//tensorflow/compiler/xla/service:transfer_manager",
248 "//tensorflow/compiler/xla/service:transfer_manager",
270 "//tensorflow/compiler/xla/service:transfer_manager",
355 "//tensorflow/compiler/xla/service:transfer_manager",
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 139 auto* transfer_manager = backend->transfer_manager(); in ExecuteAndFetchProfile() local
146 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAndFetchProfile()
148 TF_ASSERT_OK(transfer_manager->TransferLiteralToDevice( in ExecuteAndFetchProfile()
153 transfer_manager->AllocateScopedShapedBuffer( in ExecuteAndFetchProfile()
155 TF_ASSERT_OK(transfer_manager->TransferLiteralToDevice( in ExecuteAndFetchProfile()
|
| D | buffer_donation_test.cc | 101 backend_->transfer_manager()->AllocateScopedShapedBuffer( in RunAndCheck()
105 TF_CHECK_OK(backend_->transfer_manager()->TransferLiteralToDevice( in RunAndCheck()
161 backend_->transfer_manager()->TransferLiteralFromDevice( in RunAndCheck()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 142 xla::TransferManager* transfer_manager) { in FixTupleTableAsync() argument
160 return transfer_manager->WriteSingleTupleIndexTable( in FixTupleTableAsync()
286 auto transfer_manager, in UpdateDynamicInputs()
290 &runtime_input, transfer_manager)); in UpdateDynamicInputs()
467 backend->transfer_manager()->HostShapeToDeviceShape( in TPUExecute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 47 client->backend().transfer_manager()->HostShapeToDeviceShape( in AllocateShapedBuffer()
57 client->backend().transfer_manager()->GetByteSizeRequirement(subshape); in AllocateShapedBuffer()
|
| D | xla_launch_util.cc | 229 xla::TransferManager* transfer_manager = in PopulateInputs() local
230 client_->backend().transfer_manager(); in PopulateInputs()
237 transfer_manager->HostShapeToDeviceShape(shape); in PopulateInputs()
465 auto transfer_manager, in PopulateOutputs()
469 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in PopulateOutputs()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 161 auto transfer_manager, in UpdateMetadata()
173 TF_RETURN_IF_ERROR(transfer_manager->TransferArrayToDeviceAsync( in UpdateMetadata()
258 auto transfer_manager, in UpdateDynamicInputs()
261 transfer_manager->WriteTupleIndexTablesAsync(stream, shaped_buffer)); in UpdateDynamicInputs()
277 auto transfer_manager, in CreateOutputTuple()
279 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in CreateOutputTuple()
|
| /external/tensorflow/tensorflow/c/experimental/filesystem/plugins/s3/ |
| D | s3_filesystem.cc | 268 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager; member
323 auto handle = s3_file->transfer_manager->DownloadFile( in ReadS3TransferManager()
337 s3_file->transfer_manager->RetryDownload(handle); in ReadS3TransferManager()
373 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager; member
378 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager) in S3File()
382 transfer_manager(transfer_manager), in S3File()
436 auto handle = s3_file->transfer_manager->UploadFile( in Sync()
449 s3_file->transfer_manager->RetryUpload(s3_file->outfile, handle); in Sync()
|