assertEquals(null, getResponse(methodName).getResult());
  }

  /**
   * Asserts that a prior call that had caused this thread to block or wait has since returned the
   * expected boolean value.
   */
  public void assertPriorCallReturns(boolean expected, @Nullable String methodName)
      throws Exception {
    assertEquals(expected, getResponse(methodName).getResult());
  }

  /**

            throw new PluginResolutionException(
                    plugin, e.getResult().getExceptions(), logger.isDebugEnabled() ? e : null);
        } catch (DependencyResolutionException e) {
            throw new PluginResolutionException(
                    plugin, e.getResult().getCollectExceptions(), logger.isDebugEnabled() ? e : null);
        }
    }

    assertEquals(null, getResponse(methodName).getResult());
  }

  /**
   * Asserts that a prior call that had caused this thread to block or wait has since returned the
   * expected boolean value.
   */
  public void assertPriorCallReturns(boolean expected, @Nullable String methodName)
      throws Exception {
    assertEquals(expected, getResponse(methodName).getResult());
  }

  /**

            result.setDependencyGraph(node);
        } catch (DependencyCollectionException e) {
            result.setDependencyGraph(e.getResult().getRoot());
            result.setCollectionErrors(e.getResult().getExceptions());

            throw new DependencyResolutionException(
                    result,
                    "Could not collect dependencies for project " + project.getId(),

                        .append(System.lineSeparator());
            }
        }

        return msg.toString();
    }

    public DependencyResolutionResult getResult() {
        return result;
    }

    @Override
    public String getMessage() {
        return detailMessage;
    }

        Operation* parent = yield->getParentOp();
        add_to_worklist(parent->getResult(use.getOperandNumber()));
        continue;
      }
      // TODO(jpienaar): This can be generalized.
      if (isa<IdentityOp, IdentityNOp, StopGradientOp>(use.getOwner())) {
        add_to_worklist(use.getOwner()->getResult(use.getOperandNumber()));
        continue;
      }

          loc, operand_types, act, scale, mean, var, grad, op.getEpsilon(),
          feature_dim);

      x_backprop = training_op.getResult(0);

      scale_backprop = training_op.getResult(1);

      offset_backprop = training_op.getResult(2);
    } else {  // inference
      SmallVector<int64_t, 4> non_feature_dims;
      for (int64_t i = 0; i < act_type.getRank(); ++i) {

    if (isa<TF::VariantType>(elementType)) {
      // We can only replace OptionalGetValue after the inputs have been
      // replaced.
      return failure();
    }
    rewriter.replaceOpWithNewOp<TF::CastOp>(op, op.getResult(0).getType(),
                                            input);
    return success();
  }
};

class HandleOptionalNone : public OpRewritePattern<TF::OptionalNoneOp> {
 public:

  LaunchOp launch = tensorflow::WrapOpInLaunch(
      &before_cluster_builder, val_bcast.getLoc(), assigned_id, device);

  Value all_reduce =
      CreateAllReduce(replicate, inner_builder, launch.getResult(0));

  orig_to_new[val_bcast] = all_reduce;
  return success();
}

// Move all suitable broadcasts across replicas to the `cluster` into the
// `cluster`.

                        }
                    }
                }
                model = modelResult.getEffectiveModel();
            } catch (ModelBuilderException e) {
                for (ModelProblem problem : e.getResult().getProblems()) {
                    if (problem.getException() instanceof ModelResolverException) {
                        result.addException(problem.getException());