message: [XLA] Raise an error in the interpreter if args have the wrong shape.
  Otherwise, things will die somewhere deep within the HLO Evaluator, which is
  pretty confusing.
diff: @@ -68,6 +68,18 @@ StatusOr<ScopedShapedBuffer> InterpreterExecutable::ExecuteOnStream(
  "Mismatch between argument count and graph parameter count.");
  }
  + // Check that the args have the right shape.
  + for (int64 i = 0; i < computation->num_parameters(); ++i) {
  + const auto& expected_shape = computation->parameter_instruction(i)->shape();
  + const auto& actual_shape = arguments[i]->on_device_shape();
  + if (!ShapeUtil::Equal(expected_shape, actual_shape)) {
  + return InvalidArgument(
  + "Shape mismatch on parameter %d. Expected %s, but was %s.", i,
  + ShapeUtil::HumanString(expected_shape),
  + ShapeUtil::HumanString(actual_shape));
  + }
  + }
  +
  TF_ASSIGN_OR_RETURN(TransferManager * transfer_manager,
  TransferManager::GetForPlatform(platform));
  