Home
last modified time | relevance | path

Searched refs:local_client_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Doutfeed_in_nested_computation_test.cc77 comp_result = local_client_->ExecuteAndTransfer(computation, {}) in XLA_TEST_F()
84 local_client_->TransferToInfeed(LiteralUtil::CreateR0<int32_t>(1))); in XLA_TEST_F()
90 local_client_->TransferFromOutfeed(&int_shape)); in XLA_TEST_F()
96 TF_ASSERT_OK(local_client_->TransferToInfeed( in XLA_TEST_F()
103 local_client_->TransferFromOutfeed(&xfeed_shape)); in XLA_TEST_F()
111 local_client_->TransferFromOutfeed(&int_shape)); in XLA_TEST_F()
152 comp_result = local_client_->ExecuteAndTransfer(computation, {}) in XLA_TEST_F()
157 local_client_->TransferToInfeed(LiteralUtil::CreateR0<bool>(true))); in XLA_TEST_F()
160 local_client_->TransferFromOutfeed(&result_shape)); in XLA_TEST_F()
Dlocal_client_test_base.cc121 : local_client_( in LocalClientTestBase()
124 stream_executor_ = PlatformUtil::GetStreamExecutors(local_client_->platform()) in LocalClientTestBase()
125 .ValueOrDie()[local_client_->default_device_ordinal()]; in LocalClientTestBase()
127 TransferManager::GetForPlatform(local_client_->platform()).ValueOrDie(); in LocalClientTestBase()
134 return local_client_ in LiteralToShapedBuffer()
135 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal()) in LiteralToShapedBuffer()
141 return local_client_->ShapedBufferToLiteral(shaped_buffer) in ShapedBufferToLiteral()
153 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
192 local_client_->Compile(computation, argument_layouts, build_options)); in ExecuteLocally()
199 stream = local_client_->mutable_backend() in ExecuteLocally()
Dlocal_client_execute_test.cc607 for (int d = 0; d < local_client_->device_count(); ++d) { in XLA_TEST_F()
608 if (!local_client_->device_ordinal_supported(d)) { in XLA_TEST_F()
638 local_client_->device_count()), in XLA_TEST_F()
640 local_client_->device_count())); in XLA_TEST_F()
652 for (int d = 0; d < local_client_->device_count(); ++d) { in XLA_TEST_F()
653 if (!local_client_->device_ordinal_supported(d)) { in XLA_TEST_F()
657 local_client_->platform()->ExecutorForDevice(d).ValueOrDie(); in XLA_TEST_F()
716 LOG(INFO) << "default device = " << local_client_->default_device_ordinal(); in XLA_TEST_F()
718 local_client_->platform() in XLA_TEST_F()
719 ->ExecutorForDevice(local_client_->default_device_ordinal()) in XLA_TEST_F()
[all …]
Dlocal_client_allocation_test.cc45 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform()); in XLA_TEST_F()
82 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform()); in XLA_TEST_F()
83 for (int d = 0; d < local_client_->device_count(); ++d) { in XLA_TEST_F()
84 if (!local_client_->device_ordinal_supported(d)) { in XLA_TEST_F()
Dhlo_metadata_test.cc50 local_client_->Compile(builder.Build().ValueOrDie(), in TEST_F()
70 auto executable_status = local_client_->Compile( in TEST_F()
Dlocal_client_test_base.h129 LocalClient* local_client_; variable
Dtransfer_manager_test.cc47 stream_ptr_ = local_client_->mutable_backend()
58 shape, GetOrCreateAllocator(local_client_->platform()), in AllocateDeviceBuffer()
Dtest_utils_test.cc50 auto executable_status = local_client_->Compile( in XLA_TEST_F()