Home
last modified time | relevance | path

Searched refs:device_attr (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_op_device_assignment.cc45 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation() local
55 StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation() local
56 if (!device_attr) { in runOnOperation()
62 if (cloned_op_by_device.find(device_attr.getValue()) == in runOnOperation()
65 new_op->setAttr(kDeviceAttr, device_attr); in runOnOperation()
66 cloned_op_by_device[device_attr.getValue()] = new_op; in runOnOperation()
69 mlir::Operation *new_op = cloned_op_by_device[device_attr.getValue()]; in runOnOperation()
Dresource_device_inference.cc130 auto device_attr = in GetDeviceAttr() local
132 return device_attr ? device_attr.getValue() : ""; in GetDeviceAttr()
137 auto device_attr = op->getAttrOfType<mlir::StringAttr>(kDeviceAttr); in GetDeviceAttr() local
138 return device_attr ? device_attr.getValue() : ""; in GetDeviceAttr()
154 StringRef device_attr = GetDeviceAttr(func_op, arg.getArgNumber()); in ComputeResourceDevicesInComputation() local
155 if (device_attr.empty()) { in ComputeResourceDevicesInComputation()
165 auto res = AddResourceDeviceAndEmitError(arg, device_attr, func_op, result); in ComputeResourceDevicesInComputation()
180 StringRef device_attr = GetDeviceAttr(op); in ComputeResourceDevicesInComputation() local
181 if (device_attr.empty()) return WalkResult::advance(); in ComputeResourceDevicesInComputation()
183 device_attr, op, result); in ComputeResourceDevicesInComputation()
Dcross_host_transfer.cc88 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kOpDeviceAttr)) { in runOnFunction() local
89 op_device = device_attr.getValue().str(); in runOnFunction()
101 if (StringAttr device_attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction() local
103 arg_device = device_attr.getValue().str(); in runOnFunction()
107 if (StringAttr device_attr = in runOnFunction() local
109 arg_device = device_attr.getValue().str(); in runOnFunction()
Dtpu_device_propagation.cc175 auto device_attr = op_to_update->getAttrOfType<StringAttr>(kDeviceAttr); in PropagateDevicesInGraph() local
176 const bool has_device = device_attr && !device_attr.getValue().empty(); in PropagateDevicesInGraph()
177 if (has_device && tensorflow::IsTPUDevice(device_attr.getValue())) { in PropagateDevicesInGraph()
178 PopulateDeviceForOpResults(*op_to_update, device_attr.getValue(), in PropagateDevicesInGraph()
218 auto device_attr = func.getResultAttrOfType<StringAttr>( in PropagateDevicesToResults() local
220 if (device_attr && !device_attr.getValue().empty()) continue; in PropagateDevicesToResults()
Dlaunch_to_device_attribute.cc72 auto device_attr = op->getAttr(kDeviceAttr); in AssignDevicesInRegion() local
73 if (!device_attr) { in AssignDevicesInRegion()
78 if (auto device_str_attr = device_attr.dyn_cast<StringAttr>()) { in AssignDevicesInRegion()
91 << "inner op has bad 'device' attribute, got " << device_attr; in AssignDevicesInRegion()
Dreadonly_references_to_resources.cc141 StringAttr device_attr = in runOnFunction() local
143 if (!device_attr) device_attr = builder.getStringAttr(""); in runOnFunction()
155 builder.getNamedAttr("device", device_attr), in runOnFunction()
Dtf_device_assignment.cc39 if (auto device_attr = op->getAttrOfType<StringAttr>("device")) { in runOnFunction() local
41 if (device_attr.getValue() == "") { in runOnFunction()
Dcluster_formation.cc59 auto device_attr = op->getAttrOfType<StringAttr>("device"); in GetDevice() local
60 return device_attr ? device_attr.getValue() : ""; in GetDevice()
Dtpu_merge_variables_with_execute.cc129 Attribute device_attr = execute_launch.deviceAttr(); in BuildVariableAccessInfo() local
130 if (check_device && !device_attr) return infos; in BuildVariableAccessInfo()
161 if (!resource_attr || resource_attr != device_attr) continue; in BuildVariableAccessInfo()
169 if (!resource_attr || resource_attr != device_attr) continue; in BuildVariableAccessInfo()
/external/llvm-project/clang/test/Parser/
Dlambda-attr.cu7 __attribute__((device)) void device_attr() { in device_attr() function
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc394 for (const auto& device_attr : resp.local_device()) { in ListDevices() local
395 response->emplace_back(device_attr); in ListDevices()
397 for (const auto& device_attr : resp.remote_device()) { in ListDevices() local
398 response->emplace_back(device_attr); in ListDevices()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc238 if (auto device_attr = in GetArgumentNode() local
240 *node_def->mutable_device() = device_attr.getValue().str(); in GetArgumentNode()
273 if (auto device_attr = in GetReturnNode() local
275 *node_def->mutable_device() = device_attr.getValue().str(); in GetReturnNode()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_helpers.inc273 auto device_attr = op->getAttrOfType<StringAttr>("device");
274 if (!device_attr || device_attr.getValue().empty()) return true;
277 if (!DeviceNameUtils::ParseFullName(device_attr.getValue().str(), &device))
596 if (auto device_attr = op->getAttrOfType<StringAttr>("device"))
597 return device_attr.getValue();
Dtf_device.cc428 for (auto device_attr : op.devices().getValue().getValue()) { in Verify() local
429 auto device_list = device_attr.second.dyn_cast_or_null<ArrayAttr>(); in Verify()
/external/cpuinfo/test/dmesg/
Diconia-one-10.log977 [ 5.443798] (0)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
981 [ 5.444937] (0)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
1277 [ 11.419474] (0)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
1282 [ 11.420258] (0)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
Dlenovo-a6600-plus.log342 [ 2.926099] (0)[1:init][name:g_android&][g_android][USB]enable_store: device_attr->attr.name: e…
357 [ 2.950420] (0)[1:init][name:g_android&][g_android][USB]enable_store: device_attr->attr.name: e…
856 [ 5.418980] (1)[1:init][name:g_android&][g_android][USB]enable_store: device_attr->attr.name: e…
880 [ 5.419806] (1)[1:init][name:g_android&][g_android][USB]enable_store: device_attr->attr.name: e…
Dxperia-c4-dual.log636 <5>[ 8.548367]<6>.(1)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
727 <5>[ 8.602700]<6>.(0)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
Dmeizu-pro-6s.log3057 [ 3.994232] (8)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
3061 [ 3.994347] (8)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
6484 [ 23.719325] (4)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
6494 [ 23.719921] (4)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
6547 [ 23.770035] (1)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
6551 [ 23.770295] (1)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
Dmeizu-pro-7-plus.log4004 [ 7.316177] (5)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
4009 [ 7.319730] (5)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
4036 [ 7.611782] (8)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
4044 [ 7.617283] (8)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
Doppo-r9.log305 <5>[ 3.768389] (0)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
355 <5>[ 3.842269] (0)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
Dblu-r1-hd.log2672 [ 9.569388] (2)[1:init][name:g_android&][g_android][USB]enable_store: device_attr->attr.name: e…
2690 [ 9.590370] (2)[1:init][name:g_android&][g_android][USB]enable_store: device_attr->attr.name: e…
Dmeizu-pro-6.log2930 [ 3.398606] (5)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable
2934 [ 3.398801] (5)[1:init][g_android][USB]enable_store: device_attr->attr.name: enable