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.cc125 : local_client_( in LocalClientTestBase()
129 stream_executor_ = PlatformUtil::GetStreamExecutors(local_client_->platform()) in LocalClientTestBase()
133 TransferManager::GetForPlatform(local_client_->platform()).ValueOrDie(); in LocalClientTestBase()
140 return local_client_ in LiteralToShapedBuffer()
141 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal()) in LiteralToShapedBuffer()
147 return local_client_->ShapedBufferToLiteral(shaped_buffer) in ShapedBufferToLiteral()
159 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
198 local_client_->Compile(computation, argument_layouts, build_options)); in ExecuteLocally()
206 stream = local_client_->mutable_backend() in ExecuteLocally()
226 local_client_->backend().compiler()->ShapeSizeBytesFunction()); in ParseAndReturnVerifiedModule()
Dlocal_client_execute_test.cc610 for (int d = 0; d < local_client_->device_count(); ++d) { in XLA_TEST_F()
611 if (!local_client_->device_ordinal_supported(d)) { in XLA_TEST_F()
641 local_client_->device_count()), in XLA_TEST_F()
643 local_client_->device_count())); in XLA_TEST_F()
655 for (int d = 0; d < local_client_->device_count(); ++d) { in XLA_TEST_F()
656 if (!local_client_->device_ordinal_supported(d)) { in XLA_TEST_F()
660 local_client_->platform()->ExecutorForDevice(d).ValueOrDie(); in XLA_TEST_F()
719 LOG(INFO) << "default device = " << local_client_->default_device_ordinal(); in XLA_TEST_F()
721 local_client_->platform() in XLA_TEST_F()
722 ->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.cc52 local_client_->Compile(builder.Build().ValueOrDie(), in TEST_F()
76 local_client_->Compile(builder.Build().ValueOrDie(), in TEST_F()
Dlocal_client_test_base.h140 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.cc51 auto executables, local_client_->Compile(computation_status.ValueOrDie(), in XLA_TEST_F()