Home
last modified time | relevance | path

Searched defs:device_ids (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dparallel_device.py168 def device_ids(self): member in ParallelDevice
/external/tensorflow/tensorflow/core/util/autotune_maps/
Dautotune_serialize.cc120 std::vector<int> device_ids; in PopulateConvMap() local
/external/pytorch/torch/nn/parallel/
Ddistributed.py631 device_ids=None, argument
970 device_ids, argument
1647 def scatter(self, inputs, kwargs, device_ids): argument
/external/libbrillo/brillo/blkdev_utils/
Dloop_device.cc72 std::vector<std::string> device_ids = base::SplitString( in GetDeviceNumber() local
/external/pytorch/torch/csrc/distributed/c10d/
Dlogger.cpp175 const std::vector<int>& device_ids, in set_construction_data_and_log()
DTypes.hpp163 std::vector<int64_t> device_ids; member
/external/crosvm/devices/src/virtio/input/
Devdev.rs124 pub fn device_ids<T: AsRawDescriptor>(descriptor: &T) -> Result<virtio_input_device_ids> { in device_ids() function
Dmod.rs278 device_ids: virtio_input_device_ids, field
/external/google-cloud-java/java-iot/proto-google-cloud-iot-v1/src/main/proto/google/cloud/iot/v1/
Ddevice_manager.proto443 repeated string device_ids = 3; field
/external/googleapis/google/cloud/iot/v1/
Ddevice_manager.proto443 repeated string device_ids = 3; field
/external/pytorch/test/distributed/
Dtest_c10d_common.py337 device_ids, argument
362 device_ids, argument
752 device_ids, argument
Dtest_c10d_ucc.py357 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
Dtest_data_parallel.py253 def test(inner_m, dp_device, inp, device_ids, should_fail): argument
Dtest_c10d_gloo.py1499 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
Dtest_c10d_nccl.py862 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
/external/pytorch/torch/
D_utils.py757 def _get_devices_properties(device_ids): argument
/external/tensorflow/tensorflow/core/util/
Ddebug_events_writer_test.cc806 int device_ids[8]; in TEST_F() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h85 const int32_t* device_ids; member
/external/pytorch/torch/distributed/
Ddistributed_c10d.py4123 def barrier(group=GroupMember.WORLD, async_op=False, device_ids=None): argument
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py5225 device_ids, argument
/external/bcc/libbpf-tools/loongarch/
Dvmlinux_602.h51392 u32 device_ids[32]; member
114815 int device_ids[1040]; member
Dvmlinux.h51392 u32 device_ids[32]; member
114815 int device_ids[1040]; member
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h46961 u32 device_ids[32]; member
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_600.h34303 u32 device_ids[32]; member
Dvmlinux.h34303 u32 device_ids[32]; member

12