| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff.transform |
| Modifier and Type | Class and Description |
|---|---|
class |
SubGraphPredicate |
| Modifier and Type | Field and Description |
|---|---|
protected OpPredicate |
SubGraphPredicate.root |
| Modifier and Type | Field and Description |
|---|---|
protected Map<Integer,OpPredicate> |
SubGraphPredicate.opInputMatchPredicates |
protected Map<Integer,OpPredicate> |
SubGraphPredicate.opInputSubgraphPredicates |
| Modifier and Type | Method and Description |
|---|---|
static OpPredicate |
OpPredicate.classEquals(Class<?> c)
Return true if the operation class is equal to the specified class
|
static OpPredicate |
OpPredicate.nameEquals(String name)
Return true if the operation own (user specified) name equals the specified name
|
static OpPredicate |
OpPredicate.nameMatches(String regex)
Return true if the operation own (user specified) name matches the specified regular expression
|
static OpPredicate |
OpPredicate.opNameEquals(String opName)
Return true if the operation name (i.e., "add", "mul", etc - not the user specified name) equals the specified name
|
static OpPredicate |
OpPredicate.opNameMatches(String regex)
Return true if the operation name (i.e., "add", "mul", etc - not the user specified name) matches the specified regular expression
|
| Modifier and Type | Method and Description |
|---|---|
SubGraphPredicate |
SubGraphPredicate.withInputMatching(int inputNum,
@NonNull OpPredicate opPredicate)
Require the subgraph to match the specified predicate for the specified input.
Note that this does NOT add the specified input to part of the subgraph i.e., the subgraph matches if the input matches the predicate, but when returning the SubGraph itself, the function for this input is not added to the SubGraph |
SubGraphPredicate |
SubGraphPredicate.withInputSubgraph(int inputNum,
@NonNull OpPredicate opPredicate)
Require the subgraph to match the specified predicate for the specified input.
Note that this DOES add the specified input to part of the subgraph i.e., the subgraph matches if the input matches the predicate, and when returning the SubGraph itself, the function for this input IS added to the SubGraph |
static SubGraphPredicate |
SubGraphPredicate.withRoot(@NonNull OpPredicate root)
Create a SubGraphPredicate with the specified root predicate
|
| Constructor and Description |
|---|
SubGraphPredicate(OpPredicate root) |
Copyright © 2021. All rights reserved.