HomeSort by relevance Sort by last modified time
    Searched refs:transfer_manager_ (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
transfer_manager_test.cc 45 return transfer_manager_->GetByteSizeRequirement(shape);
56 return transfer_manager_
77 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal,
81 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer));
93 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal,
97 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer));
111 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal,
115 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer));
127 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal,
131 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer))
    [all...]
local_client_test_base.h 127 TransferManager* transfer_manager_; member in class:xla::LocalClientTestBase