// If device attribute is not found from the function, nothing is done.
class TFDeviceAssignmentByFuncAttrPass
    : public impl::TFDeviceAssignmentByFuncAttrPassBase<
          TFDeviceAssignmentByFuncAttrPass> {
 public:
  TFDeviceAssignmentByFuncAttrPass() = default;
  TFDeviceAssignmentByFuncAttrPass(const TFDeviceAssignmentByFuncAttrPass&) {}

  void runOnOperation() override {
    func::FuncOp func = getOperation();