/** * Constructor from parsed Keras layer configuration dictionary. * * @param layerConfig dictionary containing Keras layer configuration * @param enforceTrainingConfig whether to enforce training-related configuration options * @throws InvalidKerasConfigurationException * @throws UnsupportedKerasConfigurationException */ public KerasPoolHelper(Map<String, Object> layerConfig, boolean enforceTrainingConfig) throws InvalidKerasConfigurationException, UnsupportedKerasConfigurationException { super(layerConfig, enforceTrainingConfig); this.vertex = new PoolHelperVertex(); }
@Override public MemoryReport getMemoryReport(InputType... inputTypes) { //It's just a get op on the forward pass... no memory use InputType outputType = getOutputType(-1, inputTypes); return new LayerMemoryReport.Builder(null, PoolHelperVertex.class, inputTypes[0], outputType) .standardMemory(0, 0) //No params .workingMemory(0, 0, 0, 0).cacheMemory(0, 0) //No caching .build(); } }
@Override public PoolHelperVertex clone() { return new PoolHelperVertex(); }