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
76 "bad '{0}' attribute at index {1}, not a string", kDevicesAttr, idx)); in GetDevicesFromOp()
83 kDevicesAttr, string_attr.getValue())); in GetDevicesFromOp()
104 kDevicesAttr, name.strref())); in GetDevicesFromOp()
142 op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices)); in AddDevicesToOp()
147 auto devices_attr = op->getAttr(kDevicesAttr); in GetDevicesFromOp()
158 llvm::formatv("unsupported '{0}' attribute", kDevicesAttr)); in GetDevicesFromOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc71 constexpr char kDevicesAttr[] = "devices"; variable
445 replicate->setAttr(kDevicesAttr, builder->getDictionaryAttr(device_attrs)); in AssignDevicesToReplicate()