Searched refs:device_attrs (Results 1 – 4 of 4) sorted by relevance
35 def compute_capability_from_device_desc(device_attrs): argument52 device_attrs.physical_device_desc)
421 llvm::SmallVector<NamedAttribute, 8> device_attrs; in AssignDevicesToReplicate() local428 device_attrs.push_back( in AssignDevicesToReplicate()441 device_attrs.push_back(builder->getNamedAttr( in AssignDevicesToReplicate()445 replicate->setAttr(kDevicesAttr, builder->getDictionaryAttr(device_attrs)); in AssignDevicesToReplicate()
1161 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attrs, in FilterDevicesForRemoteWorkers() argument1163 filtered_device_mask->resize(device_attrs.size()); in FilterDevicesForRemoteWorkers()1177 for (int i = 0; i < device_attrs.size(); i++) { in FilterDevicesForRemoteWorkers()1179 DeviceNameUtils::ParseFullName(device_attrs[i].name(), &pn); in FilterDevicesForRemoteWorkers()
426 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attrs,