| /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()
115 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
131 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
147 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
167 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
187 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
201 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
220 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
235 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.h | 71 virtual StatusOr<Literal> TransferLiteralFromDevice(
74 StatusOr<Literal> TransferLiteralFromDevice( in TransferLiteralFromDevice() function
76 return TransferLiteralFromDevice(stream, device_buffer, nullptr); in TransferLiteralFromDevice()
78 virtual Status TransferLiteralFromDevice(
82 Status TransferLiteralFromDevice(se::Stream* stream, in TransferLiteralFromDevice() function
85 return TransferLiteralFromDevice(stream, device_buffer, literal, nullptr); in TransferLiteralFromDevice()
100 virtual void TransferLiteralFromDevice(
104 void TransferLiteralFromDevice(se::Stream* stream, in TransferLiteralFromDevice() function
108 return TransferLiteralFromDevice(stream, device_buffer, literal, done, in TransferLiteralFromDevice()
|
| D | transfer_manager.cc | 47 StatusOr<Literal> TransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::TransferManager
60 TransferLiteralFromDevice( in TransferLiteralFromDevice()
74 Status TransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::TransferManager
84 TransferLiteralFromDevice( in TransferLiteralFromDevice()
197 return TransferLiteralFromDevice(stream, shaped_buffer, literal, in TransferArrayFromDevice()
|
| D | hlo_runner.cc | 132 StatusOr<Literal> HloRunner::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::HloRunner
136 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in TransferLiteralFromDevice()
152 return TransferLiteralFromDevice(result); in Execute()
183 return TransferLiteralFromDevice(result); in Execute()
409 backend().transfer_manager()->TransferLiteralFromDevice( in ExecuteReplicated()
|
| D | generic_transfer_manager.h | 43 void TransferLiteralFromDevice(
|
| D | hlo_runner.h | 110 StatusOr<Literal> TransferLiteralFromDevice(const ShapedBuffer& buffer);
|
| D | generic_transfer_manager.cc | 58 void GenericTransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::GenericTransferManager
|
| D | service.cc | 80 transfer_manager->TransferLiteralFromDevice(stream, *argument)); in RecordArguments()
92 transfer_manager->TransferLiteralFromDevice(stream, result)); in RecordResult()
959 execute_backend_->transfer_manager()->TransferLiteralFromDevice( in TransferToClient()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 216 return backend_->transfer_manager()->TransferLiteralFromDevice(stream.get(), in LiteralFromShapedBuffer()
282 return backend().transfer_manager()->TransferLiteralFromDevice(stream.get(), in ShapedBufferToLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 91 transfer_manager->TransferLiteralFromDevice( in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 247 transfer_manager_->TransferLiteralFromDevice( in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 237 return transfer_manager->TransferLiteralFromDevice(stream.get(), in ToLiteral()
|