| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 88 auto transfer_manager = backend->transfer_manager(); in AllocateScopedShapedBuffer() local
101 xla::Shape on_device_shape = transfer_manager->HostShapeToDeviceShape(shape); in AllocateScopedShapedBuffer()
114 uint64 size = transfer_manager->GetByteSizeRequirement(subshape); in AllocateScopedShapedBuffer()
125 transfer_manager->WriteTupleIndexTables(stream.get(), *(buffer->get()))); in AllocateScopedShapedBuffer()
186 auto transfer_manager = backend->transfer_manager(); in CreateAndTransfer() local
193 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in CreateAndTransfer()
224 auto transfer_manager = backend->transfer_manager(); in ToLiteral() local
237 return transfer_manager->TransferLiteralFromDevice(stream.get(), in ToLiteral()
249 auto transfer_manager = backend->transfer_manager(); in WriteLiteral() local
251 return transfer_manager->TransferLiteralToDevice(stream.get(), literal, in WriteLiteral()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 216 return backend_->transfer_manager()->TransferLiteralFromDevice(stream.get(), in LiteralFromShapedBuffer()
269 backend().transfer_manager()->AllocateScopedShapedBuffer( in LiteralToShapedBuffer()
273 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in LiteralToShapedBuffer()
282 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in ShapedBufferToLiteral()
295 return backend().transfer_manager()->TransferLiteralToInfeed(executor, in TransferToInfeedLocal()
304 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralFromOutfeed( in TransferFromOutfeedLocal()
319 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferToLocalServer()
323 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferToLocalServer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 83 TF_ASSIGN_OR_RETURN(TransferManager * transfer_manager, in ExecuteOnStream()
91 transfer_manager->TransferLiteralFromDevice( in ExecuteOnStream()
107 transfer_manager->AllocateScopedShapedBuffer( in ExecuteOnStream()
110 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in ExecuteOnStream()
|
| D | BUILD | 16 "//tensorflow/compiler/xla/service:transfer_manager",
93 "//tensorflow/compiler/xla/service:transfer_manager",
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 111 TF_ASSIGN_OR_RETURN(auto transfer_manager, in CreateBackend()
116 new Backend(platform, compiler, stream_executors, transfer_manager, in CreateBackend()
145 TransferManager* transfer_manager, in Backend() argument
150 transfer_manager_(transfer_manager), in Backend()
|
| D | hlo_runner.cc | 100 backend().transfer_manager()->AllocateScopedShapedBuffer( in TransferLiteralToDevice()
105 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferLiteralToDevice()
136 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in TransferLiteralFromDevice()
306 backend().transfer_manager()->AllocateScopedShapedBuffer( in ExecuteReplicated()
308 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in ExecuteReplicated()
337 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralToInfeed( in ExecuteReplicated()
356 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in ExecuteReplicated()
409 backend().transfer_manager()->TransferLiteralFromDevice( in ExecuteReplicated()
|
| D | service.cc | 74 se::Stream* stream, TransferManager* transfer_manager, in RecordArguments() argument
80 transfer_manager->TransferLiteralFromDevice(stream, *argument)); in RecordArguments()
88 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult() argument
92 transfer_manager->TransferLiteralFromDevice(stream, result)); in RecordResult()
701 execute_backend_->transfer_manager(), in ExecuteGraphParallel()
748 execute_backend_->transfer_manager(), in ExecuteGraphParallel()
899 execute_backend_->transfer_manager(), executable->hlo_snapshot())); in Execute()
915 execute_backend_->transfer_manager(), in Execute()
959 execute_backend_->transfer_manager()->TransferLiteralFromDevice( in TransferToClient()
991 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,
|
| D | transfer_manager.h | 257 TransferManagerCreationFunction transfer_manager);
|
| D | BUILD | 696 ":transfer_manager",
737 ":transfer_manager",
964 name = "transfer_manager",
965 srcs = ["transfer_manager.cc"],
966 hdrs = ["transfer_manager.h"],
991 ":transfer_manager",
2159 ":transfer_manager",
3488 ":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 | dynamic_ops_test.cc | 742 auto* transfer_manager = in BM_DynamicSlice() local
769 .transfer_manager() in BM_DynamicSlice()
774 ASSERT_IS_OK(transfer_manager->TransferLiteralToDevice( in BM_DynamicSlice()
|
| D | local_client_execute_test.cc | 906 auto* transfer_manager = in BM_LocalClientOverhead() local
918 transfer_manager in BM_LocalClientOverhead()
925 transfer_manager->TransferLiteralToDevice(stream.get(), literal, buffer)); in BM_LocalClientOverhead()
|
| D | tuple_test.cc | 575 TF_EXPECT_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in XLA_TEST_F()
|
| D | BUILD | 77 "//tensorflow/compiler/xla/service:transfer_manager",
266 "//tensorflow/compiler/xla/service:transfer_manager",
1097 "//tensorflow/compiler/xla/service:transfer_manager",
1962 "//tensorflow/compiler/xla/service:transfer_manager",
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 51 client->backend().transfer_manager()->HostShapeToDeviceShape( in AllocateShapedBuffer()
61 client->backend().transfer_manager()->GetByteSizeRequirement(subshape); in AllocateShapedBuffer()
|
| D | xla_device_context.h | 81 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
|
| D | xla_device_context.cc | 84 transfer_manager_(client->backend().transfer_manager()), in XlaDeviceContext()
|
| D | xla_launch_util.cc | 239 client_->backend().transfer_manager()->HostShapeToDeviceShape(shape); in PopulateInputs()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | developing_new_backend.md | 72 …nsferManager`](https://www.tensorflow.org/code/tensorflow/compiler/xla/service/transfer_manager.h):
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 397 "//tensorflow/compiler/xla/service:transfer_manager",
795 "//tensorflow/compiler/xla/service:transfer_manager",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 48 "//tensorflow/compiler/xla/service:transfer_manager",
|