| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transfer_manager_test.cc | 81 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
97 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
119 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
137 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
155 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
173 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
189 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
209 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
229 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
243 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
[all …]
|
| D | buffer_donation_test.cc | 161 backend_->transfer_manager()->TransferLiteralFromDevice( in RunAndCheck()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.h | 77 virtual StatusOr<Literal> TransferLiteralFromDevice(
80 StatusOr<Literal> TransferLiteralFromDevice( in TransferLiteralFromDevice() function
82 return TransferLiteralFromDevice(stream, device_buffer, nullptr); in TransferLiteralFromDevice()
84 virtual Status TransferLiteralFromDevice(
88 Status TransferLiteralFromDevice(se::Stream* stream, in TransferLiteralFromDevice() function
91 return TransferLiteralFromDevice(stream, device_buffer, literal, nullptr); in TransferLiteralFromDevice()
106 virtual void TransferLiteralFromDevice(
110 void TransferLiteralFromDevice(se::Stream* stream, in TransferLiteralFromDevice() function
114 return TransferLiteralFromDevice(stream, device_buffer, literal, done, in TransferLiteralFromDevice()
|
| D | transfer_manager.cc | 50 StatusOr<Literal> TransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::TransferManager
63 TransferLiteralFromDevice( in TransferLiteralFromDevice()
77 Status TransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::TransferManager
87 TransferLiteralFromDevice( in TransferLiteralFromDevice()
198 return TransferLiteralFromDevice(stream, shaped_buffer, literal, in TransferArrayFromDevice()
|
| D | hlo_runner.cc | 83 StatusOr<Literal> HloRunner::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::HloRunner
87 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in TransferLiteralFromDevice()
103 return TransferLiteralFromDevice(result.Result()); in Execute()
116 return TransferLiteralFromDevice(result.Result()); in ExecuteWithExecutable()
312 backend().transfer_manager()->TransferLiteralFromDevice( in ExecuteReplicatedImpl()
|
| D | generic_transfer_manager.h | 43 void TransferLiteralFromDevice(
|
| D | hlo_runner.h | 64 StatusOr<Literal> TransferLiteralFromDevice(const ShapedBuffer& buffer);
|
| D | generic_transfer_manager.cc | 61 void GenericTransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::GenericTransferManager
|
| D | service.cc | 81 transfer_manager->TransferLiteralFromDevice(stream, *argument)); in RecordArguments()
93 transfer_manager->TransferLiteralFromDevice(stream, result)); in RecordResult()
936 execute_backend_->transfer_manager()->TransferLiteralFromDevice( in TransferToClient()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 207 backend->transfer_manager()->TransferLiteralFromDevice( in DumpArguments()
226 backend->transfer_manager()->TransferLiteralFromDevice( in DumpOutputsAndSaveSnapshot()
419 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in ShapedBufferToLiteral()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_transfer_manager.h | 49 void TransferLiteralFromDevice(
|
| D | tpu_transfer_manager.cc | 190 void TpuTransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in tensorflow::tpu::TpuTransferManager
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.cc | 113 transfer_manager->TransferLiteralFromDevice( in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 257 transfer_manager_->TransferLiteralFromDevice( in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 710 transfer_manager->TransferLiteralFromDevice( in DoWork()
775 transfer_manager->TransferLiteralFromDevice( in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 254 return transfer_manager->TransferLiteralFromDevice(stream.get(), in StoreToLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1247 ->TransferLiteralFromDevice(stream, shaped_buffer, literal, in ToLiteral()
|