@Override public UnstackVertex clone() { return new UnstackVertex(from, stackSize); }
@Override public MemoryReport getMemoryReport(InputType... inputTypes) { //Get op with dup - accounted for in activations size (no working memory) //Do one dup on the forward pass (output activations). Accounted for in output activations. InputType outputType = getOutputType(-1, inputTypes); return new LayerMemoryReport.Builder(null, UnstackVertex.class, inputTypes[0], outputType).standardMemory(0, 0) //No params .workingMemory(0, 0, 0, 0).cacheMemory(0, 0) //No caching .build(); } }