Package | Description |
---|---|
org.nd4j.linalg.api.ops | |
org.nd4j.linalg.api.ops.custom | |
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 |
---|---|
CustomOpDescriptor |
ScatterUpdate.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 © 2020. All rights reserved.