Searched refs:default_device_ordinal (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 38 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()
|
| D | local_client.h | 176 int default_device_ordinal() const;
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 716 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()
|
| D | local_client_test_base.cc | 125 .ValueOrDie()[local_client_->default_device_ordinal()]; in LocalClientTestBase()
135 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal()) in LiteralToShapedBuffer()
|
| D | xla_hlo_profile_test.cc | 142 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()
|
| D | dynamic_ops_test.cc | 744 int device_ordinal = client->default_device_ordinal(); in BM_DynamicSlice()
|
| D | fusion_test.cc | 842 int device_ordinal = client->default_device_ordinal(); in BM_ParallelFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 102 backend().default_device_ordinal())); in TransferLiteralToDevice()
209 GetServiceRunOptionsForDevice(backend().default_device_ordinal(), &stream, in ExecuteWithDeviceBuffers()
246 GetServiceRunOptionsForDevice(backend().default_device_ordinal(), &stream, in ExecuteWithDeviceBuffers()
|
| D | backend.h | 102 int default_device_ordinal() const;
|
| D | backend.cc | 174 int Backend::default_device_ordinal() const { in default_device_ordinal() function in xla::Backend
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 123 build_options.set_device_ordinal(client->default_device_ordinal()); in Compile()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 154 : client_->default_device_ordinal()); in BuildExecutable()
|
| D | xla_launch_util.cc | 253 client_->platform(), client_->default_device_ordinal()); in PopulateInputs()
|