| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 99 int device_count() const { return stream_executors_.size(); } in device_count() function
|
| D | platform_util.cc | 214 int device_count = platform->VisibleDeviceCount(); in GetStreamExecutors() local
|
| /external/skia/src/compute/common/cl/ |
| D | find_cl.c | 87 cl_uint device_count; in clFindIdsByName() local
|
| /external/skqp/src/compute/common/cl/ |
| D | find_cl.c | 87 cl_uint device_count; in clFindIdsByName() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_resolver_local_test.cc | 36 auto* device_count = options.config.mutable_device_count(); in DeviceResolverLocalTest() local
|
| D | collective_executor_mgr_test.cc | 38 auto* device_count = options.config.mutable_device_count(); in CollectiveExecutorMgrTest() local
|
| D | collective_rma_local_test.cc | 43 auto* device_count = options.config.mutable_device_count(); in CollectiveRemoteAccessLocalTest() local
|
| D | collective_param_resolver_local_test.cc | 38 auto* device_count = options.config.mutable_device_count(); in CollectiveParamResolverLocalTest() local
|
| D | function_threadpool_test.cc | 55 auto* device_count = options.config.mutable_device_count(); in Init() local
|
| D | partitioning_utils_test.cc | 43 auto* device_count = options.config.mutable_device_count(); in SetUp() local
|
| D | hierarchical_tree_broadcaster.cc | 130 int device_count = 0; in InitializeCollectiveParams() local
|
| D | process_function_library_runtime_test.cc | 70 auto* device_count = options.config.mutable_device_count(); in Init() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed_test.cc | 206 const int device_count = num_workers * num_devices; in IssueRequests() local
223 int device_count = num_workers * num_devices; in IssueRequest() local
249 int device_count = num_workers * num_devices; in ValidateCollectiveParams() local
|
| D | rpc_collective_executor_mgr_test.cc | 43 auto* device_count = options.config.mutable_device_count(); in RpcCollectiveExecutorMgrTest() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | platform.cc | 110 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op_test.cc | 60 auto* device_count = options.config.mutable_device_count(); in Init() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 55 void TestWithDeviceCount(const int device_count) { in TestWithDeviceCount()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 224 int LocalClient::device_count() const { in device_count() function in xla::LocalClient
|
| D | client.cc | 366 int64 device_count) { in GetDeviceHandles()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 145 auto* device_count = options.config.mutable_device_count(); in InitFunctionLibraryRuntime() local
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | profile_analyzer_cli.py | 477 self, device_name, device_index, device_count, argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 320 map<string, int32> device_count = 1; field
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 309 int64 device_count = 1; field
|
| D | xla_data.proto | 313 int64 device_count = 2; field
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 80 def device_count(self): member in Backend
129 def device_count(self): member in XlaLocalBackend
166 def device_count(self): member in XrtBackend
|