Searched defs:device_attr (Results 1 – 13 of 13) sorted by relevance
88 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kOpDeviceAttr)) { in runOnFunction() local101 if (StringAttr device_attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction() local107 if (StringAttr device_attr = in runOnFunction() local
45 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation() local55 StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation() local
130 auto device_attr = in GetDeviceAttr() local137 auto device_attr = op->getAttrOfType<mlir::StringAttr>(kDeviceAttr); in GetDeviceAttr() local154 StringRef device_attr = GetDeviceAttr(func_op, arg.getArgNumber()); in ComputeResourceDevicesInComputation() local180 StringRef device_attr = GetDeviceAttr(op); in ComputeResourceDevicesInComputation() local
39 if (auto device_attr = op->getAttrOfType<StringAttr>("device")) { in runOnFunction() local
175 auto device_attr = op_to_update->getAttrOfType<StringAttr>(kDeviceAttr); in PropagateDevicesInGraph() local218 auto device_attr = func.getResultAttrOfType<StringAttr>( in PropagateDevicesToResults() local
72 auto device_attr = op->getAttr(kDeviceAttr); in AssignDevicesInRegion() local
141 StringAttr device_attr = in runOnFunction() local
59 auto device_attr = op->getAttrOfType<StringAttr>("device"); in GetDevice() local
129 Attribute device_attr = execute_launch.deviceAttr(); in BuildVariableAccessInfo() local
7 __attribute__((device)) void device_attr() { in device_attr() function
394 for (const auto& device_attr : resp.local_device()) { in ListDevices() local397 for (const auto& device_attr : resp.remote_device()) { in ListDevices() local
238 if (auto device_attr = in GetArgumentNode() local273 if (auto device_attr = in GetReturnNode() local
428 for (auto device_attr : op.devices().getValue().getValue()) { in Verify() local