OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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