Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Ddistributed.py631 device_ids=None, argument
970 device_ids, argument
1647 def scatter(self, inputs, kwargs, device_ids): argument
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_common.py337 device_ids, argument
362 device_ids, argument
752 device_ids, argument
H A Dtest_c10d_ucc.py357 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
H A Dtest_data_parallel.py253 def test(inner_m, dp_device, inp, device_ids, should_fail): argument
H A Dtest_c10d_gloo.py1499 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
H A Dtest_c10d_nccl.py862 self, devices, device_ids, multi_device=False, gradient_as_bucket_view=False argument
/aosp_15_r20/device/google/cuttlefish/host/frontend/webrtc_operator/assets/js/
Dindex.js50 #ShowNewDeviceList(device_ids) { argument
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/
H A Dparallel_device.py168 def device_ids(self): member in ParallelDevice
/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dautotune_serialize.cc120 std::vector<int> device_ids; in PopulateConvMap() local
/aosp_15_r20/external/libbrillo/brillo/blkdev_utils/
H A Dloop_device.cc72 std::vector<std::string> device_ids = base::SplitString( in GetDeviceNumber() local
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dlogger.cpp175 const std::vector<int>& device_ids, in set_construction_data_and_log()
H A DTypes.hpp163 std::vector<int64_t> device_ids; member
/aosp_15_r20/external/crosvm/devices/src/virtio/input/
H A Devdev.rs124 pub fn device_ids<T: AsRawDescriptor>(descriptor: &T) -> Result<virtio_input_device_ids> { in device_ids() function
H A Dmod.rs253 device_ids: virtio_input_device_ids, field
/aosp_15_r20/hardware/interfaces/tv/input/1.0/vts/functional/
H A DVtsHalTvInputV1_0TargetTest.cpp238 std::vector<int32_t> device_ids; in TEST_P() local
/aosp_15_r20/hardware/interfaces/tv/input/aidl/vts/functional/
H A DVtsHalTvInputTargetTest.cpp209 vector<int32_t> device_ids; in TEST_P() local
/aosp_15_r20/external/google-cloud-java/java-iot/proto-google-cloud-iot-v1/src/main/proto/google/cloud/iot/v1/
H A Ddevice_manager.proto443 repeated string device_ids = 3; field
/aosp_15_r20/external/googleapis/google/cloud/iot/v1/
H A Ddevice_manager.proto443 repeated string device_ids = 3; field
/aosp_15_r20/external/pytorch/torch/
H A D_utils.py757 def _get_devices_properties(device_ids): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Ddebug_events_writer_test.cc806 int device_ids[8]; in TEST_F() local
/aosp_15_r20/system/keymint/wire/src/
H A Dlegacy.rs488 pub device_ids: Vec<u8>, field
/aosp_15_r20/device/google/cuttlefish/host/commands/assemble_cvd/
Dflags.cc810 std::vector<std::string> device_ids; in CreateNumToWebrtcDeviceIdMap() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h85 const int32_t* device_ids; member
/aosp_15_r20/tools/netsim/rust/daemon/src/devices/
H A Ddevices_handler.rs751 let mut device_ids = Vec::new(); in reset_all() localVariable
/aosp_15_r20/external/pytorch/torch/distributed/
H A Ddistributed_c10d.py4123 def barrier(group=GroupMember.WORLD, async_op=False, device_ids=None): argument

12