Home
last modified time | relevance | path

Searched refs:kDevicesAttr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddevice_util.cc39 constexpr char kDevicesAttr[] = "tf.devices"; variable
73 "bad '{0}' attribute at index {1}, not a string", kDevicesAttr, idx)); in GetDevicesFromOp()
80 kDevicesAttr, string_attr.getValue())); in GetDevicesFromOp()
101 kDevicesAttr, name.strref())); in GetDevicesFromOp()
139 op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices)); in AddDevicesToOp()
144 auto devices_attr = op->getAttr(kDevicesAttr); in GetDevicesFromOp()
155 llvm::formatv("unsupported '{0}' attribute", kDevicesAttr)); in GetDevicesFromOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc66 constexpr char kDevicesAttr[] = "devices"; variable
433 replicate->setAttr(kDevicesAttr, builder->getDictionaryAttr(device_attrs)); in AssignDevicesToReplicate()