| Package | Description |
|---|---|
| org.nd4j.linalg.api.ops.executioner |
| Modifier and Type | Method and Description |
|---|---|
TadPack |
DefaultOpExecutioner.tadShapeInfoAndOffsets(INDArray array,
int[] dimension) |
TadPack |
OpExecutioner.tadShapeInfoAndOffsets(INDArray array,
int[] dimension)
This method returns host/device tad buffers
|
Copyright © 2021. All rights reserved.