| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff | |
| org.nd4j.autodiff.samediff.array | |
| org.nd4j.autodiff.samediff.optimize | |
| org.nd4j.autodiff.samediff.optimize.optimizations |
| Modifier and Type | Method and Description |
|---|---|
void |
ArrayHolder.initFrom(ArrayHolder arrayHolder)
Initialize from the specified array holder.
|
void |
SameDiff.setArrayHolders(@NonNull ArrayHolder variableArrayHolder,
@NonNull ArrayHolder constantArrayHolder,
boolean initialize)
Set the array holders for variable and constant arrays
NOTE: this is usually reserved for developers and internal use, and should not be needed by almost all users See ArrayHolder for more details |
void |
SameDiff.setArrayHolders(@NonNull ArrayHolder variableArrayHolder,
@NonNull ArrayHolder constantArrayHolder,
boolean initialize)
Set the array holders for variable and constant arrays
NOTE: this is usually reserved for developers and internal use, and should not be needed by almost all users See ArrayHolder for more details |
| Modifier and Type | Class and Description |
|---|---|
class |
OptimizedGraphArrayHolder |
class |
SingleThreadArrayHolder |
class |
ThreadSafeArrayHolder |
| Modifier and Type | Method and Description |
|---|---|
void |
ThreadSafeArrayHolder.initFrom(ArrayHolder arrayHolder) |
void |
SingleThreadArrayHolder.initFrom(ArrayHolder arrayHolder) |
void |
OptimizedGraphArrayHolder.initFrom(ArrayHolder arrayHolder) |
| Constructor and Description |
|---|
OptimizedGraphArrayHolder(ArrayHolder underlyingHolder) |
| Modifier and Type | Method and Description |
|---|---|
boolean |
Optimizer.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
| Modifier and Type | Method and Description |
|---|---|
boolean |
ShapeFunctionOptimizations.FuseChainedPermutes.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
boolean |
ShapeFunctionOptimizations.FuseChainedReshapes.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
boolean |
ShapeFunctionOptimizations.FuseChainedConcatOps.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
boolean |
ConstantFunctionOptimizations.FoldConstantFunctions.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
boolean |
CuDNNFunctionOptimizations.CudnnConv2dNCHWtoNHWCConversion.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
boolean |
UnusedFunctionOptimizations.RemoveUnusedConstants.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
boolean |
IdentityFunctionOptimizations.RemoveIdentityPermute.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
boolean |
IdentityFunctionOptimizations.RemoveIdentityOps.checkAndApply(SameDiff sd,
OptimizationHelper helper,
SameDiffOp op,
ArrayHolder constantArrays,
ArrayHolder variablesArrays) |
Copyright © 2022. All rights reserved.