Home
last modified time | relevance | path

Searched refs:TransferLiteralToDevice (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dtransfer_manager_test.cc77 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()
115 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
133 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
151 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
169 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
185 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
205 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
225 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
239 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
[all …]
Dxla_hlo_profile_test.cc148 TF_ASSERT_OK(transfer_manager->TransferLiteralToDevice( in ExecuteAndFetchProfile()
155 TF_ASSERT_OK(transfer_manager->TransferLiteralToDevice( in ExecuteAndFetchProfile()
Dbuffer_donation_test.cc105 TF_CHECK_OK(backend_->transfer_manager()->TransferLiteralToDevice( in RunAndCheck()
Ddynamic_ops_test.cc791 ASSERT_IS_OK(transfer_manager->TransferLiteralToDevice( in BM_DynamicSlice()
Dlocal_client_execute_test.cc974 transfer_manager->TransferLiteralToDevice(stream.get(), literal, buffer)); in BM_LocalClientOverhead()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h129 virtual Status TransferLiteralToDevice(
133 Status TransferLiteralToDevice(se::Stream* stream, in TransferLiteralToDevice() function
136 return TransferLiteralToDevice(stream, literal, device_buffer, nullptr); in TransferLiteralToDevice()
Dhlo_runner.cc48 StatusOr<ScopedShapedBuffer> HloRunner::TransferLiteralToDevice( in TransferLiteralToDevice() function in xla::HloRunner
56 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferLiteralToDevice()
67 TransferLiteralToDevice(*literal)); in TransferLiteralsToDevice()
242 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in ExecuteReplicatedImpl()
Dhlo_runner.h59 StatusOr<ScopedShapedBuffer> TransferLiteralToDevice(const Literal& literal);
Dtransfer_manager.cc98 Status TransferManager::TransferLiteralToDevice( in TransferLiteralToDevice() function in xla::TransferManager
175 return TransferLiteralToDevice(stream, literal, shaped_buffer, in TransferArrayToDeviceAsync()
Dservice.cc973 execute_backend_->transfer_manager()->TransferLiteralToDevice( in TransferToServer()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc179 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in CreateAndTransfer()
275 return transfer_manager->TransferLiteralToDevice(stream.get(), literal, in WriteLiteral()
314 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in SwapIn()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc133 TF_RETURN_IF_ERROR(transfer_manager->TransferLiteralToDevice( in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc410 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in LiteralToShapedBuffer()
457 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralToDevice( in TransferToLocalServer()