HomeSort by relevance Sort by last modified time
    Searched full:local_client_ (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_test_base.cc 120 : local_client_(
123 stream_executor_ = PlatformUtil::GetStreamExecutors(local_client_->platform())
124 .ValueOrDie()[local_client_->default_device_ordinal()];
126 TransferManager::GetForPlatform(local_client_->platform()).ValueOrDie();
133 return local_client_
134 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal())
140 return local_client_->ShapedBufferToLiteral(shaped_buffer)
152 local_client_->backend().inter_op_thread_pool());
154 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform()));
195 local_client_->Compile(computation, argument_layouts, build_options))
    [all...]
local_client_allocation_test.cc 40 ComputationBuilder builder(local_client_, TestName());
45 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform());
76 ComputationBuilder builder(local_client_, TestName());
82 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform());
83 for (int d = 0; d < local_client_->device_count(); ++d) {
84 if (!local_client_->device_ordinal_supported(d)) {
hlo_metadata_test.cc 43 ComputationBuilder builder(local_client_, "add");
51 local_client_->Compile(builder.Build().ValueOrDie(),
64 ComputationBuilder builder(local_client_, "add");
71 auto executable_status = local_client_->Compile(
local_client_execute_test.cc 59 ComputationBuilder builder(local_client_, TestName());
70 ComputationBuilder builder(local_client_, TestName());
84 ComputationBuilder builder(local_client_, TestName());
98 ComputationBuilder builder(local_client_, TestName());
113 ComputationBuilder builder(local_client_, TestName());
132 ComputationBuilder builder(local_client_, TestName());
165 ComputationBuilder builder(local_client_, TestName());
202 ComputationBuilder builder(local_client_, TestName());
230 ComputationBuilder builder(local_client_, TestName());
264 ComputationBuilder builder(local_client_, TestName())
    [all...]
local_client_test_base.h 134 LocalClient* local_client_; member in class:xla::LocalClientTestBase
transfer_manager_test.cc 51 shape, GetOrCreateAllocator(local_client_->platform()),

Completed in 219 milliseconds