Uses of Class
org.nd4j.autodiff.samediff.transform.OpPredicate
-
Packages that use OpPredicate Package Description org.nd4j.autodiff.samediff.transform -
-
Uses of OpPredicate in org.nd4j.autodiff.samediff.transform
Subclasses of OpPredicate in org.nd4j.autodiff.samediff.transform Modifier and Type Class Description classSubGraphPredicateFields in org.nd4j.autodiff.samediff.transform declared as OpPredicate Modifier and Type Field Description protected OpPredicateSubGraphPredicate. rootFields in org.nd4j.autodiff.samediff.transform with type parameters of type OpPredicate Modifier and Type Field Description protected Map<Integer,OpPredicate>SubGraphPredicate. opInputMatchPredicatesprotected Map<Integer,OpPredicate>SubGraphPredicate. opInputSubgraphPredicatesMethods in org.nd4j.autodiff.samediff.transform that return OpPredicate Modifier and Type Method Description static OpPredicateOpPredicate. classEquals(Class<?> c)Return true if the operation class is equal to the specified classstatic OpPredicateOpPredicate. nameEquals(String name)Return true if the operation own (user specified) name equals the specified namestatic OpPredicateOpPredicate. nameMatches(String regex)Return true if the operation own (user specified) name matches the specified regular expressionstatic OpPredicateOpPredicate. opNameEquals(String opName)Return true if the operation name (i.e., "add", "mul", etc - not the user specified name) equals the specified namestatic OpPredicateOpPredicate. opNameMatches(String regex)Return true if the operation name (i.e., "add", "mul", etc - not the user specified name) matches the specified regular expressionMethods in org.nd4j.autodiff.samediff.transform with parameters of type OpPredicate Modifier and Type Method Description SubGraphPredicateSubGraphPredicate. 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 SubGraphSubGraphPredicateSubGraphPredicate. 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 SubGraphstatic SubGraphPredicateSubGraphPredicate. withRoot(@NonNull OpPredicate root)Create a SubGraphPredicate with the specified root predicateConstructors in org.nd4j.autodiff.samediff.transform with parameters of type OpPredicate Constructor Description SubGraphPredicate(OpPredicate root)
-