Home
last modified time | relevance | path

Searched defs:num_devices (Results 1 – 25 of 85) sorted by relevance

1234

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc116 void DefineWorkers(int num_workers, int num_devices, in DefineWorkers()
125 int num_devices, bool nccl) { in DefineWorker()
163 void DefineCollectiveParams(int num_workers, int num_devices, in DefineCollectiveParams()
180 CollectiveParams* CreateCollectiveParams(int num_workers, int num_devices, in CreateCollectiveParams()
200 void IssueRequests(int num_workers, int num_devices) { in IssueRequests()
236 void ValidateCollectiveParams(int num_workers, int num_devices) { in ValidateCollectiveParams()
281 void RestartWorker(int worker_idx, int num_workers, int num_devices, in RestartWorker()
323 const int num_devices = 1; in TEST_F() local
332 const int num_devices = 2; in TEST_F() local
341 const int num_devices = 1; in TEST_F() local
[all …]
/external/OpenCL-CTS/test_conformance/d3d10/
Dmain.cpp63 cl_uint num_devices = 0; in main() local
90 …ation(cl_platform_id platform, IDXGIAdapter* pAdapter, ID3D10Device* pDevice, cl_uint* num_devices) in TestAdapterEnumeration()
210 cl_uint num_devices = 0; in TestAdapterDevices() local
/external/OpenCL-CTS/test_conformance/d3d11/
Dmain.cpp66 cl_uint num_devices = 0; in main() local
98 cl_uint* num_devices) in TestAdapterEnumeration()
208 cl_uint num_devices = 0; in TestAdapterDevices() local
/external/OpenCL-CTS/test_conformance/buffers/
Dtest_buffer_migrate.cpp51 …migrations migrate, cl_command_queue *queues, cl_mem *mem_objects, cl_uint num_devices, cl_mem_mig… in migrateMemObject()
81 static cl_int restoreBuffer(cl_command_queue *queues, cl_mem *buffers, cl_uint num_devices, cl_mem_… in restoreBuffer()
105 cl_uint num_devices, num_devices_limited; in test_buffer_migrate() local
Dtest_image_migrate.cpp62 cl_uint num_devices, cl_mem_migration_flags *flags, MTdata d) in migrateMemObject()
93 static cl_int restoreImage(cl_command_queue *queues, cl_mem *mem_objects, cl_uint num_devices, in restoreImage()
122 cl_uint num_devices, num_devices_limited; in test_image_migrate() local
/external/OpenCL-CTS/test_conformance/SVM/
Dtest_fine_grain_memory_consistency.cpp68 … &context, clCommandQueueWrapper* queues, clKernelWrapper &kernel, cl_uint num_devices, cl_uint nu… in launch_kernels_and_verify()
147 cl_uint num_devices = 0; in test_svm_fine_grain_memory_consistency() local
Dtest_shared_address_space_fine_grain.cpp27 cl_uint num_devices = 0; in test_svm_shared_address_space_fine_grain() local
Dtest_allocate_shared_buffer.cpp48 cl_uint num_devices = 0; in test_svm_allocate_shared_buffer() local
Dtest_fine_grain_sync_buffers.cpp51 cl_uint num_devices = 0; in test_svm_fine_grain_sync_buffers() local
Dtest_pointer_passing.cpp42 cl_uint num_devices = 0; in test_svm_pointer_passing() local
Dtest_set_kernel_exec_info_svm_ptrs.cpp49 cl_uint num_devices = 0; in test_svm_set_kernel_exec_info_svm_ptrs() local
Dtest_shared_address_space_fine_grain_buffers.cpp68 cl_uint num_devices = 0; in test_svm_shared_address_space_fine_grain_buffers() local
Dtest_byte_granularity.cpp59 cl_uint num_devices = 0; in test_svm_byte_granularity() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpermuter_test.cc50 void Init(int num_workers, int num_devices, in Init()
86 int num_devices, int tensor_len, int fail_after) { in RunTest()
Dring_gatherer_test.cc47 void Init(int num_workers, int num_devices, DataType dtype, in Init()
80 int num_devices, int num_subdivs, int tensor_len, in RunTest()
Dhierarchical_tree_broadcaster_test.cc138 void Init(int num_workers, int num_devices, DataType dtype, in Init()
170 int num_devices, int tensor_len, int fail_after, in RunTest()
/external/webrtc/modules/audio_device/
Daudio_device_unittest.cc723 int num_devices = audio_device()->PlayoutDevices(); in TEST_P() local
740 int num_devices = audio_device()->RecordingDevices(); in TEST_P() local
756 int num_devices = audio_device()->PlayoutDevices(); in TEST_P() local
779 int num_devices = audio_device()->RecordingDevices(); in TEST_P() local
817 int num_devices = audio_device()->PlayoutDevices(); in TEST_P() local
844 int num_devices = audio_device()->RecordingDevices(); in TEST_P() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc36 int num_devices = node->num_inputs(); in ReplaceReduce() local
121 int num_devices = 0; // Number of distinct devices, incremented below. in ReplaceBroadcast() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreplicated_io_feed_test.cc31 static DeviceAssignment MakeDeviceAssn(size_t num_devices) { in MakeDeviceAssn()
/external/tensorflow/tensorflow/lite/nnapi/
Dnnapi_util.cc55 uint32_t num_devices = 0; in GetDeviceNamesList() local
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py129 def split(cls, tensor, split_dimension, num_devices, input_shape=None): argument
300 num_devices, argument
/external/OpenCL-CTS/test_conformance/api/
Dtest_platform.cpp181 cl_uint num_devices; in test_get_platform_ids() local
/external/virglrenderer/src/
Dvrend_winsys_egl.c174 static EGLint virgl_egl_find_3d_device(struct gbm_device_info *dev_infos, EGLint num_devices, uint3… in virgl_egl_find_3d_device()
187 static EGLint virgl_egl_find_matching_device(struct gbm_device_info *dev_infos, EGLint num_devices,… in virgl_egl_find_matching_device()
200 EGLint num_devices = 0; in virgl_egl_get_device() local
/external/mesa3d/src/loader/
Dloader.c139 int i, num_devices, fd = -1; in loader_open_render_node() local
306 int i, num_devices, fd = -1; in loader_get_user_preferred_fd() local
/external/blktrace/iowatcher/
Dtracers.c52 int start_blktrace(char **devices, int num_devices, char *trace_name, char *dest) in start_blktrace()

1234