public class DeviceTADManager extends BasicTADManager
Modifier and Type | Field and Description |
---|---|
protected List<Map<TadDescriptor,org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer>>> |
tadCache |
allocator, nativeOps
Constructor and Description |
---|
DeviceTADManager() |
Modifier and Type | Method and Description |
---|---|
org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer> |
getTADOnlyShapeInfo(INDArray array,
int[] dimension) |
protected List<Map<TadDescriptor,org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer>>> tadCache
public org.apache.commons.math3.util.Pair<DataBuffer,DataBuffer> getTADOnlyShapeInfo(INDArray array, int[] dimension)
getTADOnlyShapeInfo
in interface TADManager
getTADOnlyShapeInfo
in class BasicTADManager
Copyright © 2016. All Rights Reserved.