Home
last modified time | relevance | path

Searched refs:default_device_ordinal (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc38 device_ordinal = backend->default_device_ordinal(); in BorrowStreamForDevice()
110 : backend_->default_device_ordinal(); in ValidateExecutionOptions()
232 int LocalClient::default_device_ordinal() const { in default_device_ordinal() function in xla::LocalClient
233 return local_service_->backend().default_device_ordinal(); in default_device_ordinal()
250 updated_options.set_device_ordinal(default_device_ordinal()); in Compile()
Dlocal_client.h176 int default_device_ordinal() const;
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc716 LOG(INFO) << "default device = " << local_client_->default_device_ordinal(); in XLA_TEST_F()
719 ->ExecutorForDevice(local_client_->default_device_ordinal()) in XLA_TEST_F()
788 literal, local_client_->default_device_ordinal(), allocator_)); in XLA_TEST_F()
828 literal, local_client_->default_device_ordinal(), allocator_)); in XLA_TEST_F()
863 local_client_->default_device_ordinal())); in XLA_TEST_F()
887 local_client_->default_device_ordinal())); in XLA_TEST_F()
891 shape, local_client_->default_device_ordinal())); in XLA_TEST_F()
908 int device_ordinal = client->default_device_ordinal(); in BM_LocalClientOverhead()
Dlocal_client_test_base.cc125 .ValueOrDie()[local_client_->default_device_ordinal()]; in LocalClientTestBase()
135 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal()) in LiteralToShapedBuffer()
Dxla_hlo_profile_test.cc142 backend->BorrowStream(backend->default_device_ordinal())); in ExecuteAndFetchProfile()
147 lhs_arg_shape, allocator, backend->default_device_ordinal())); in ExecuteAndFetchProfile()
154 rhs_arg_shape, allocator, backend->default_device_ordinal())); in ExecuteAndFetchProfile()
Ddynamic_ops_test.cc744 int device_ordinal = client->default_device_ordinal(); in BM_DynamicSlice()
Dfusion_test.cc842 int device_ordinal = client->default_device_ordinal(); in BM_ParallelFusion()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc102 backend().default_device_ordinal())); in TransferLiteralToDevice()
209 GetServiceRunOptionsForDevice(backend().default_device_ordinal(), &stream, in ExecuteWithDeviceBuffers()
246 GetServiceRunOptionsForDevice(backend().default_device_ordinal(), &stream, in ExecuteWithDeviceBuffers()
Dbackend.h102 int default_device_ordinal() const;
Dbackend.cc174 int Backend::default_device_ordinal() const { in default_device_ordinal() function in xla::Backend
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc123 build_options.set_device_ordinal(client->default_device_ordinal()); in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.cc154 : client_->default_device_ordinal()); in BuildExecutable()
Dxla_launch_util.cc253 client_->platform(), client_->default_device_ordinal()); in PopulateInputs()