public Iterator<MatrixSlice> iterateNonEmpty() { final int[] keys = rowVectors.keySet().toIntArray(); return new AbstractIterator<MatrixSlice>() { private int slice; @Override protected MatrixSlice computeNext() { if (slice >= rowVectors.size()) { return endOfData(); } int i = keys[slice]; Vector row = rowVectors.get(i); slice++; return new MatrixSlice(row, i); } }; }
/** special method necessary for efficient serialization */ public IntArrayList nonZeroRowIndices() { return new IntArrayList(rowVectors.keySet().toIntArray()); }
@Override public void reset() { iter = map.keySet().iterator(); } };
@Override public Iterator<IntWritable> fastKeyIterator() { return new ReusableIterator<IntIterator>(map.keySet().iterator()) { @Override public IntWritable next() { reusableKey.set(iter.nextInt()); return reusableKey; } @Override public void reset() { iter = map.keySet().iterator(); } }; }
public Iterator<MatrixSlice> iterateNonEmpty() { final int[] keys = rowVectors.keySet().toIntArray(); return new AbstractIterator<MatrixSlice>() { private int slice; @Override protected MatrixSlice computeNext() { if (slice >= rowVectors.size()) { return endOfData(); } int i = keys[slice]; Vector row = rowVectors.get(i); slice++; return new MatrixSlice(row, i); } }; }
/** special method necessary for efficient serialization */ public IntArrayList nonZeroRowIndices() { return new IntArrayList(rowVectors.keySet().toIntArray()); }
@Override public Iterable<IntWritable> getPartitionDestinationVertices( int partitionId) { Int2ObjectOpenHashMap<DataInputOutput> partitionMap = map.get(partitionId); List<IntWritable> vertices = Lists.newArrayListWithCapacity(partitionMap.size()); IntIterator iterator = partitionMap.keySet().iterator(); while (iterator.hasNext()) { vertices.add(new IntWritable(iterator.nextInt())); } return vertices; }
@Override public void finalizeStore() { for (int partitionId : nascentMap.keySet()) { // nascent vertices are present only in nascent map map.get(partitionId).putAll(nascentMap.get(partitionId)); } nascentMap.clear(); }
private void serializeMap(final DataOutput out, final Map map) throws IOException { Class oCls = map.getClass(); if (oCls.equals(Int2ObjectOpenHashMap.class)) { serialize(out, ((Int2ObjectOpenHashMap) map).keySet().toIntArray()); serialize(out, map.values().toArray()); } else if (oCls.equals(Float2ObjectOpenHashMap.class)) {
private void serializeMap(final DataOutput out, final Map map) throws IOException { Class oCls = map.getClass(); if (oCls.equals(Int2ObjectOpenHashMap.class)) { serialize(out, ((Int2ObjectOpenHashMap) map).keySet().toIntArray()); serialize(out, map.values().toArray()); } else if (oCls.equals(Float2ObjectOpenHashMap.class)) {
final int layerSize = inputBuff.capacity(); if (layersCountToFree > 0) { for (final int i : layersCache.keySet()) { layer = layersCache.remove(i); if (layer != null) {
final int layerSize = inputBuff.capacity(); if(layersCountToFree > 0) { for(final int i : layersCache.keySet()) { layer = layersCache.remove(i); if(layer != null) {