Searched refs:default_device_ordinal (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 118 ->Allocate(backend_->default_device_ordinal(), host_buffer.size()) in RunMlirModuleWithHostBuffers()
121 auto stream = backend_->BorrowStream(backend_->default_device_ordinal()) in RunMlirModuleWithHostBuffers()
168 auto stream = backend_->BorrowStream(backend_->default_device_ordinal()) in CompileMlirText()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 38 device_ordinal = backend->default_device_ordinal(); in BorrowStreamForDevice()
85 : backend_->default_device_ordinal(); in ValidateExecutionOptions()
344 int LocalClient::default_device_ordinal() const { in default_device_ordinal() function in xla::LocalClient
345 return local_service_->backend().default_device_ordinal(); in default_device_ordinal()
362 updated_options.set_device_ordinal(default_device_ordinal()); in Compile()
|
| D | local_client.h | 200 int default_device_ordinal() const;
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 719 LOG(INFO) << "default device = " << local_client_->default_device_ordinal(); in XLA_TEST_F()
722 ->ExecutorForDevice(local_client_->default_device_ordinal()) in XLA_TEST_F()
839 literal, local_client_->default_device_ordinal(), allocator_)); in XLA_TEST_F()
879 literal, local_client_->default_device_ordinal(), allocator_)); in XLA_TEST_F()
914 local_client_->default_device_ordinal())); in XLA_TEST_F()
938 local_client_->default_device_ordinal())); in XLA_TEST_F()
942 local_client_->default_device_ordinal(), &result)); in XLA_TEST_F()
957 int device_ordinal = client->default_device_ordinal(); in BM_LocalClientOverhead()
|
| 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 | local_client_test_base.cc | 141 ->LiteralToShapedBuffer(literal, local_client_->default_device_ordinal()) in LiteralToShapedBuffer()
|
| D | dynamic_ops_test.cc | 761 int device_ordinal = client->default_device_ordinal(); in BM_DynamicSlice()
|
| D | cpu_gpu_fusion_test.cc | 841 int device_ordinal = client->default_device_ordinal(); in BM_ParallelFusion()
|
| D | dot_operation_test.cc | 1823 int device_ordinal = client->default_device_ordinal(); in DOT_ReorderContracting()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 102 int default_device_ordinal() const;
|
| D | backend.cc | 147 int Backend::default_device_ordinal() const { in default_device_ordinal() function in xla::Backend
|
| D | hlo_runner.cc | 53 backend().default_device_ordinal())); in TransferLiteralToDevice()
166 GetServiceRunOptionsForDevice(backend().default_device_ordinal(), &stream, in ExecuteWithDeviceBuffers()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 57 client, allocator, client->default_device_ordinal(), in Run()
|
| D | xla_compilation_cache.cc | 171 : client_->default_device_ordinal()); in BuildExecutable()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 256 : client->default_device_ordinal(); in Compute()
474 : closure.client()->default_device_ordinal(); in Compute()
|