Searched refs:num_devices_ (Results 1 – 2 of 2) sorted by relevance
| /external/OpenCL-CTS/test_conformance/events/ |
| D | test_userevents.cpp | 56 cl_uint num_devices_;\
57 clGetProgramInfo(program,CL_PROGRAM_NUM_DEVICES,sizeof(num_devices_),&num_devices_,NULL);\
59 device_list=(cl_device_id *)malloc(num_devices_*sizeof(cl_device_id));\
60 clGetProgramInfo(program,CL_PROGRAM_DEVICES,num_devices_*sizeof(cl_device_id),device_list,NULL);\
61 for (unsigned i=0;i<num_devices_;++i) {\
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 46 OP_REQUIRES_OK(c, c->GetAttr("num_devices", &num_devices_)); in NcclAsyncOpBase()
56 int num_devices() const { return num_devices_; } in num_devices()
59 int num_devices_; member in tensorflow::__anonc894cd950111::NcclAsyncOpBase
|