Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc138 void DefineWorkers(int num_workers, int num_devices, in DefineWorkers()
147 int num_devices, bool nccl) { in DefineWorker()
178 void DefineCollectiveParams(int num_workers, int num_devices, in DefineCollectiveParams()
191 CollectiveParams* CreateCollectiveParams(int num_workers, int num_devices, in CreateCollectiveParams()
208 void IssueRequests(int num_workers, int num_devices) { in IssueRequests()
244 void ValidateCollectiveParams(int num_workers, int num_devices) { in ValidateCollectiveParams()
289 void RestartWorker(int worker_idx, int num_workers, int num_devices, in RestartWorker()
327 const int num_devices = 1; in TEST_F() local
336 const int num_devices = 2; in TEST_F() local
345 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/llvm-project/openmp/libomptarget/test/api/
Domp_get_num_devices.c12 int num_devices = omp_get_num_devices(); in test_omp_get_num_devices() local
/external/llvm-project/openmp/runtime/test/api/
Domp_get_num_devices.c8 int num_devices = omp_get_num_devices(); in test_omp_get_num_devices() 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/llvm-project/openmp/libomptarget/test/offloading/
Dd2d_memcpy.c16 const int num_devices = omp_get_num_devices(); in main() local
/external/tensorflow/tensorflow/lite/nnapi/
Dnnapi_util.cc51 uint32_t num_devices = 0; in GetDeviceNamesList() local
/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/experimental/xla_sharding/
Dxla_sharding.py129 def split(cls, tensor, split_dimension, num_devices, input_shape=None): argument
300 num_devices, argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute_options.py139 num_devices = options.create_option( variable in DistributeOptions
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h60 int num_devices; in initializeDeviceProp() local
119 int num_devices; in initializeDeviceProp() local
/external/OpenCL-CTS/test_conformance/api/
Dtest_platform.cpp181 cl_uint num_devices; in test_get_platform_ids() local
/external/blktrace/iowatcher/
Dtracers.c52 int start_blktrace(char **devices, int num_devices, char *trace_name, char *dest) in start_blktrace()
/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

1234