code: @Override
  	public void prepare() throws Exception {
  		final TaskConfig config = this.taskContext.getTaskConfig();
  		if (config.getDriverStrategy() != DriverStrategy.ALL_REDUCE) {
  			throw new Exception("Unrecognized driver strategy for AllReduce driver: " + config.getDriverStrategy().name());
  		}
  		
  		TypeSerializerFactory<T> serializerFactory = this.taskContext.getInputSerializer(0);
  		this.serializer = serializerFactory.getSerializer();
  		this.input = this.taskContext.getInput(0);
  	}
docs: --------------------------------------------------------------------------------------------