| /external/llvm-project/openmp/libomptarget/plugins/cuda/src/ |
| D | rtl.cpp | 147 void resizeStreamPool(const int DeviceId, const size_t NewSize) { in resizeStreamPool() argument
148 std::vector<CUstream> &Pool = StreamPool[DeviceId]; in resizeStreamPool()
152 CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); in resizeStreamPool()
215 CUstream getStream(const int DeviceId) { in getStream() argument
216 const std::lock_guard<std::mutex> Lock(*StreamMtx[DeviceId]); in getStream()
217 int &Id = NextStreamId[DeviceId]; in getStream()
219 if (Id == StreamPool[DeviceId].size()) { in getStream()
221 resizeStreamPool(DeviceId, Id * 2); in getStream()
223 return StreamPool[DeviceId][Id++]; in getStream()
240 void returnStream(const int DeviceId, CUstream Stream) { in returnStream() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.h | 39 class DeviceId {
41 DeviceId(DeviceId&&) = default;
42 DeviceId(const DeviceId&) = default;
43 DeviceId& operator=(const DeviceId&) = default;
45 bool operator==(const DeviceId& other) const { return id() == other.id(); }
46 bool operator!=(const DeviceId& other) const { return !(*this == other); }
51 explicit DeviceId(int id) : id_(id) {} in DeviceId() function
62 void Insert(DeviceId device_id);
83 if (!func(DeviceId(word_index * kWordSize + bit_index))) { in ForEach()
114 bool IsGpu(DeviceId device) const { return is_gpu_[device.id()]; } in IsGpu()
[all …]
|
| D | device_util.cc | 26 void DeviceSet::Insert(DeviceId device_id) { in Insert()
51 xla::StatusOr<DeviceId> DeviceInfoCache::GetIdFor(absl::string_view name) { in GetIdFor()
68 name_to_id_.emplace(string(name), DeviceId(new_id)); in GetIdFor()
77 return DeviceId(new_id); in GetIdFor()
82 device_set.ForEach([&](DeviceId device_id) { in DebugString()
100 xla::StatusOr<absl::optional<jit::DeviceId>> PickDeviceForXlaImpl( in PickDeviceForXlaImpl()
113 absl::optional<jit::DeviceId> maybe_gpu_device; in PickDeviceForXlaImpl()
114 absl::optional<jit::DeviceId> maybe_cpu_device; in PickDeviceForXlaImpl()
115 absl::optional<jit::DeviceId> maybe_unknown_device; in PickDeviceForXlaImpl()
125 [&](const jit::DeviceId& d0, absl::optional<jit::DeviceId>* d1) -> bool { in PickDeviceForXlaImpl()
[all …]
|
| D | device_util_test.cc | 30 TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, cache.GetIdFor(name)); in PickDeviceHelper()
35 jit::DeviceId result_id, in PickDeviceHelper()
118 TF_ASSERT_OK_AND_ASSIGN(jit::DeviceId device_id, in SimpleRoundTripTestForDeviceSet()
124 device_set.ForEach([&](jit::DeviceId device_id) { in SimpleRoundTripTestForDeviceSet()
|
| D | mark_for_compilation_pass.cc | 62 using jit::DeviceId;
129 absl::optional<DeviceId> resource_op_device, in Cluster()
183 const absl::optional<DeviceId>& resource_op_device() const { in resource_op_device()
232 absl::optional<DeviceId> resource_op_device_;
319 absl::optional<DeviceId> resource_op_device, in MakeNewCluster()
1030 TF_ASSIGN_OR_RETURN(DeviceId device, in BuildInitialClusterSet()
1034 absl::optional<DeviceId> resource_op_device; in BuildInitialClusterSet()
1599 absl::optional<jit::DeviceId> maybe_chosen_device, in AreDevicesCompatible()
1606 jit::DeviceId chosen_device = *maybe_chosen_device; in AreDevicesCompatible()
1615 [&](absl::optional<DeviceId> resource_op_device) { in AreDevicesCompatible()
[all …]
|
| D | build_xla_ops_pass.cc | 287 jit::DeviceId device, bool* result) { in DeviceRequiresCompilation()
346 xla::StatusOr<jit::DeviceId> InferDeviceForCluster( in InferDeviceForCluster()
357 TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, in InferDeviceForCluster()
367 TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, in InferDeviceForCluster()
372 TF_ASSIGN_OR_RETURN(jit::DeviceId result, in InferDeviceForCluster()
476 jit::DeviceId device, in ReplaceNodeWithXlaCompileAndXlaRun()
|
| /external/angle/src/libANGLE/renderer/ |
| D | driver_utils.cpp | 92 bool IsSandyBridge(uint32_t DeviceId) in IsSandyBridge() argument
94 return std::find(std::begin(SandyBridge), std::end(SandyBridge), DeviceId) != in IsSandyBridge()
98 bool IsIvyBridge(uint32_t DeviceId) in IsIvyBridge() argument
100 return std::find(std::begin(IvyBridge), std::end(IvyBridge), DeviceId) != std::end(IvyBridge); in IsIvyBridge()
103 bool IsHaswell(uint32_t DeviceId) in IsHaswell() argument
105 return std::find(std::begin(Haswell), std::end(Haswell), DeviceId) != std::end(Haswell); in IsHaswell()
108 bool IsBroadwell(uint32_t DeviceId) in IsBroadwell() argument
110 return std::find(std::begin(Broadwell), std::end(Broadwell), DeviceId) != std::end(Broadwell); in IsBroadwell()
113 bool IsCherryView(uint32_t DeviceId) in IsCherryView() argument
115 return std::find(std::begin(CherryView), std::end(CherryView), DeviceId) != in IsCherryView()
[all …]
|
| D | driver_utils.h | 154 bool IsSandyBridge(uint32_t DeviceId);
155 bool IsIvyBridge(uint32_t DeviceId);
156 bool IsHaswell(uint32_t DeviceId);
157 bool IsBroadwell(uint32_t DeviceId);
158 bool IsCherryView(uint32_t DeviceId);
159 bool IsSkylake(uint32_t DeviceId);
160 bool IsBroxton(uint32_t DeviceId);
161 bool IsKabylake(uint32_t DeviceId);
162 bool Is9thGenIntel(uint32_t DeviceId);
|
| /external/llvm-project/openmp/libomptarget/src/ |
| D | omptarget.cpp | 870 uint64_t getLoopTripCount(int64_t DeviceId) { in getLoopTripCount() argument
871 DeviceTy &Device = PM->Devices[DeviceId]; in getLoopTripCount()
1057 int processDataBefore(int64_t DeviceId, void *HostPtr, int32_t ArgNum, in processDataBefore() argument
1064 DeviceTy &Device = PM->Devices[DeviceId]; in processDataBefore()
1178 int processDataAfter(int64_t DeviceId, void *HostPtr, int32_t ArgNum, in processDataAfter() argument
1184 DeviceTy &Device = PM->Devices[DeviceId]; in processDataAfter()
1211 int target(int64_t DeviceId, void *HostPtr, int32_t ArgNum, void **ArgBases, in target() argument
1215 DeviceTy &Device = PM->Devices[DeviceId]; in target()
1229 assert(TM->Table->TargetsTable.size() > (size_t)DeviceId && in target()
1231 TargetTable = TM->Table->TargetsTable[DeviceId]; in target()
[all …]
|
| D | private.h | 36 extern int target(int64_t DeviceId, void *HostPtr, int32_t ArgNum,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_placer.cc | 124 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId() function in xla::ComputationPlacer
140 DeviceId(replica, computation, replica_count, computation_count)); in AssignDevices()
|
| D | tpu_computation_placer.h | 35 StatusOr<int> DeviceId(int replica, int computation, int replica_count,
|
| D | tpu_computation_placer.cc | 37 StatusOr<int> TpuComputationPlacer::DeviceId(int replica, int computation, in DeviceId() function in tensorflow::tpu::TpuComputationPlacer
|
| D | computation_placer.h | 81 virtual StatusOr<int> DeviceId(int replica, int computation,
|
| D | local_service.cc | 191 return backend().computation_placer()->DeviceId( in ReplicaNumberToDeviceOrdinal()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_tensor_handle.h | 50 virtual int DeviceId(Status* status) const = 0;
|
| /external/libusb/libusb/os/ |
| D | windows_usbdk.h | 63 PUSB_DK_DEVICE_ID DeviceId
|
| /external/angle/src/gpu_info_util/ |
| D | SystemInfo_win.cpp | 72 device.deviceId = desc.DeviceId; in GetDevicesFromDXGI()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.cc | 50 int CustomDeviceTensorHandle::DeviceId(Status* status) const { in DeviceId() function in tensorflow::CustomDeviceTensorHandle
|
| D | custom_device.h | 97 int DeviceId(Status* status) const override;
|
| /external/mesa3d/src/gallium/targets/d3dadapter9/ |
| D | drm.c | 149 get_bus_info(fd, &drvid->VendorId, &drvid->DeviceId, in read_descriptor()
166 drvid->DeviceId = fallback_cards[i].device_id; in read_descriptor()
|
| D | description.c | 242 drvid->DeviceId = fallback_dev; in d3d_match_vendor_id()
|
| /external/angle/src/feature_support_util/ |
| D | feature_support_util_unittest.cpp | 133 "DeviceId" : 234, in TEST_F()
|
| /external/oss-fuzz/projects/nginx/fuzz/ |
| D | http_request_fuzzer.dict | 134 "X-ATT-DeviceId"
|
| /external/llvm-project/openmp/libomptarget/plugins/amdgpu/src/ |
| D | rtl.cpp | 568 int32_t dataRetrieve(int32_t DeviceId, void *HstPtr, void *TgtPtr, int64_t Size, in dataRetrieve() argument
571 assert(DeviceId < DeviceInfo.NumberOfDevices && "Device ID too large"); in dataRetrieve()
581 DeviceId); in dataRetrieve()
595 int32_t dataSubmit(int32_t DeviceId, void *TgtPtr, void *HstPtr, int64_t Size, in dataSubmit() argument
599 assert(DeviceId < DeviceInfo.NumberOfDevices && "Device ID too large"); in dataSubmit()
608 DeviceId); in dataSubmit()
|