Searched refs:TransferLiteralToDevice (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transfer_manager_test.cc | 77 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
93 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
111 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
127 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
143 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
163 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
183 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
197 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
216 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
231 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
[all …]
|
| D | xla_hlo_profile_test.cc | 148 TF_ASSERT_OK(transfer_manager->TransferLiteralToDevice( in ExecuteAndFetchProfile()
155 TF_ASSERT_OK(transfer_manager->TransferLiteralToDevice( in ExecuteAndFetchProfile()
|
| D | dynamic_ops_test.cc | 774 ASSERT_IS_OK(transfer_manager->TransferLiteralToDevice( in BM_DynamicSlice()
|
| D | local_client_execute_test.cc | 925 transfer_manager->TransferLiteralToDevice(stream.get(), literal, buffer)); in BM_LocalClientOverhead()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.h | 123 virtual Status TransferLiteralToDevice(
127 Status TransferLiteralToDevice(se::Stream* stream, in TransferLiteralToDevice() function
130 return TransferLiteralToDevice(stream, literal, device_buffer, nullptr); in TransferLiteralToDevice()
|
| D | hlo_runner.cc | 97 StatusOr<ScopedShapedBuffer> HloRunner::TransferLiteralToDevice( in TransferLiteralToDevice() function in xla::HloRunner
105 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferLiteralToDevice()
116 TransferLiteralToDevice(*literal)); in TransferLiteralsToDevice()
308 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in ExecuteReplicated()
|
| D | transfer_manager.cc | 95 Status TransferManager::TransferLiteralToDevice( in TransferLiteralToDevice() function in xla::TransferManager
173 return TransferLiteralToDevice(stream, literal, shaped_buffer, in TransferArrayToDeviceAsync()
|
| D | hlo_runner.h | 105 StatusOr<ScopedShapedBuffer> TransferLiteralToDevice(const Literal& literal);
|
| D | service.cc | 996 execute_backend_->transfer_manager()->TransferLiteralToDevice( in TransferToServer()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 273 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in LiteralToShapedBuffer()
323 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferToLocalServer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 110 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 193 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in CreateAndTransfer()
251 return transfer_manager->TransferLiteralToDevice(stream.get(), literal, in WriteLiteral()
|