Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc145 void DefineWorkers(int num_workers, int num_devices, in DefineWorkers()
158 const string& device_type, int num_devices) { in DefineWorker()
183 void DefineCollectiveParams(int num_workers, int num_devices) { in DefineCollectiveParams()
205 void IssueRequests(int num_workers, int num_devices) { in IssueRequests()
222 void IssueRequest(int num_workers, int num_devices, int idx) { in IssueRequest()
248 void ValidateCollectiveParams(int num_workers, int num_devices) { in ValidateCollectiveParams()
299 const int num_devices = 1; in TEST_F() local
308 const int num_devices = 2; in TEST_F() local
317 const int num_devices = 3; in TEST_F() local
Ddevice_resolver_distributed_test.cc144 void DefineWorkers(int num_workers, int num_devices, in DefineWorkers()
153 int num_devices) { in DefineWorker()
Dcollective_rma_distributed_test.cc184 const int num_devices = 1; in SetUp() local
213 const string& device_type, int num_devices) { in DefineWorker()
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py92 def split(cls, tensor, split_dimension, num_devices): argument
184 def split(tensor, split_dimension, num_devices): argument
/external/webrtc/webrtc/tools/e2e_quality/audio/
Daudio_e2e_harness.cc70 int num_devices = 0; in RunHarness() 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/python/grappler/
Dcontroller.py86 def num_devices(self): member in Controller
/external/u-boot/test/dm/
Dtest-fdt.c132 int dm_check_devices(struct unit_test_state *uts, int num_devices) in dm_check_devices()
169 const int num_devices = 7; in dm_test_fdt() local
Dbus.c107 int num_devices = 7; in dm_test_bus_children() local
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h60 int num_devices; in initializeDeviceProp() local
119 int num_devices; in initializeDeviceProp() local
/external/blktrace/iowatcher/
Dtracers.c52 int start_blktrace(char **devices, int num_devices, char *trace_name, char *dest) in start_blktrace()
Dblkparse.h66 int num_devices; member
/external/webrtc/talk/media/devices/
Dmacdevicemanager.cc121 size_t num_devices = propsize / sizeof(AudioDeviceID); in GetAudioDeviceIDs() local
/external/kernel-headers/original/uapi/linux/
Dpktcdvd.h101 __u32 num_devices; /* out: Largest device index + 1 */ member
/external/mesa3d/src/gallium/state_trackers/clover/api/
Dcontext.cpp71 cl_uint num_devices; in clCreateContextFromType() local
/external/webrtc/webrtc/tools/agc/
Dagc_harness.cc158 int num_devices = 0; in PrintDevices() local
/external/mesa3d/src/loader/
Dloader.c163 int i, num_devices, fd; in loader_get_user_preferred_fd() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc183 const int num_devices = TF_DeviceListCount(devices); in GetDeviceName() local
/external/u-boot/net/
Deth_legacy.c239 int num_devices = 0; in eth_initialize() local
Deth-uclass.c387 int num_devices = 0; in eth_initialize() local
/external/tensorflow/tensorflow/core/kernels/
Dnccl_ops.cc52 int num_devices() const { return num_devices_; } in num_devices() function in tensorflow::__anon49c74ebe0111::NcclAsyncOpBase
/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.cc338 const int32 num_devices = in Launched() local
Dring_gatherer_test.cc133 void Init(int num_workers, int num_devices, DataType dtype, in Init()
263 int num_devices, int num_subdivs, int tensor_len, in RunTest()
Dring_reducer_test.cc155 void Init(int num_workers, int num_devices, DataType dtype, in Init()
285 int num_devices, int num_subdivs, int tensor_len, in RunTest()
/external/u-boot/fs/btrfs/
Dctree.h123 __u64 num_devices; member
160 __u64 num_devices; member

12