| Package | Description |
|---|---|
| org.nd4j.linalg.api.ops | |
| org.nd4j.linalg.api.ops.executioner |
| Modifier and Type | Method and Description |
|---|---|
CustomOpDescriptor |
CustomOp.getDescriptor()
Get the custom op descriptor if one is available.
|
CustomOpDescriptor |
DynamicCustomOp.getDescriptor() |
| Modifier and Type | Method and Description |
|---|---|
Map<String,CustomOpDescriptor> |
DefaultOpExecutioner.getCustomOperations() |
Map<String,CustomOpDescriptor> |
OpExecutioner.getCustomOperations()
This method returns names of all custom operations available in current backend, and their number of input/output arguments
|
Copyright © 2021. All rights reserved.