| Modifier and Type | Class and Description |
|---|---|
class |
DynamicCustomOp |
class |
NoOp |
| Modifier and Type | Method and Description |
|---|---|
CustomOp |
BaseOp.toCustomOp() |
CustomOp |
Op.toCustomOp()
|
| Modifier and Type | Class and Description |
|---|---|
class |
CompatSparseToDense |
class |
CompatStringSplit |
| Modifier and Type | Class and Description |
|---|---|
class |
DecodeBitmap |
class |
DecodeThreshold |
class |
EncodeBitmap |
class |
EncodeThreshold |
| Modifier and Type | Class and Description |
|---|---|
class |
AdjustContrast |
class |
AdjustHue |
class |
AdjustSaturation |
class |
BarnesEdgeForces |
class |
BarnesHutGains |
class |
BarnesHutSymmetrize |
class |
BetaInc |
class |
BitCast |
class |
CompareAndBitpack |
class |
Digamma |
class |
DivideNoNan |
class |
DrawBoundingBoxes |
class |
FakeQuantWithMinMaxVarsPerChannel |
class |
Flatten |
class |
FusedBatchNorm |
class |
HsvToRgb |
class |
Igamma |
class |
Igammac |
class |
KnnMinDistance |
class |
Lgamma |
class |
LinearSolve |
class |
Logdet |
class |
Lstsq |
class |
Lu |
class |
MatrixBandPart |
class |
Polygamma |
class |
RandomCrop |
class |
RgbToGrayscale |
class |
RgbToHsv |
class |
RgbToYiq |
class |
RgbToYuv |
class |
Roll |
class |
SpTreeCell |
class |
ToggleBits |
class |
Tri |
class |
TriangularSolve |
class |
Triu |
class |
TriuBp |
class |
YiqToRgb |
class |
YuvToRgb |
| Modifier and Type | Method and Description |
|---|---|
CustomOp |
DefaultOpExecutioner.execAndReturn(CustomOp op) |
CustomOp |
OpExecutioner.execAndReturn(CustomOp op)
This method executes given CustomOp
PLEASE NOTE: You're responsible for input/output validation
|
| Modifier and Type | Method and Description |
|---|---|
INDArray[] |
DefaultOpExecutioner.allocateOutputArrays(CustomOp op) |
INDArray[] |
OpExecutioner.allocateOutputArrays(CustomOp op)
Equivalent to calli
|
List<LongShapeDescriptor> |
DefaultOpExecutioner.calculateOutputShape(CustomOp op) |
List<LongShapeDescriptor> |
OpExecutioner.calculateOutputShape(CustomOp op) |
List<LongShapeDescriptor> |
DefaultOpExecutioner.calculateOutputShape(CustomOp op,
OpContext opContext) |
List<LongShapeDescriptor> |
OpExecutioner.calculateOutputShape(CustomOp op,
OpContext opContext) |
static void |
OpExecutionerUtil.checkForInf(CustomOp op,
OpContext oc) |
static void |
OpExecutionerUtil.checkForNaN(CustomOp op,
OpContext oc) |
protected void |
DefaultOpExecutioner.checkForWorkspaces(CustomOp op,
OpContext oc) |
INDArray[] |
DefaultOpExecutioner.exec(CustomOp op) |
INDArray[] |
OpExecutioner.exec(CustomOp op) |
INDArray[] |
DefaultOpExecutioner.exec(CustomOp op,
OpContext context) |
INDArray[] |
OpExecutioner.exec(CustomOp op,
OpContext context)
This method executes op with given context
|
CustomOp |
DefaultOpExecutioner.execAndReturn(CustomOp op) |
CustomOp |
OpExecutioner.execAndReturn(CustomOp op)
This method executes given CustomOp
PLEASE NOTE: You're responsible for input/output validation
|
long |
DefaultOpExecutioner.profilingConfigurableHookIn(CustomOp op,
OpContext oc) |
void |
DefaultOpExecutioner.profilingConfigurableHookOut(CustomOp op,
OpContext oc,
long timeStart) |
long |
DefaultOpExecutioner.profilingHookIn(CustomOp op,
OpContext oc)
Deprecated.
|
void |
DefaultOpExecutioner.profilingHookOut(CustomOp op,
OpContext oc,
long timeStart)
Deprecated.
|
| Modifier and Type | Class and Description |
|---|---|
class |
BiasAdd
Bias addition gradient operation.
|
class |
BiasAddGrad |
class |
BroadcastTo |
| Modifier and Type | Class and Description |
|---|---|
class |
Select |
class |
Where |
class |
WhereNumpy |
| Modifier and Type | Class and Description |
|---|---|
class |
BaseCompatOp |
class |
Enter |
class |
Exit |
class |
LoopCond |
class |
Merge |
class |
NextIteration |
class |
StopGradient |
class |
Switch |
class |
While |
| Modifier and Type | Class and Description |
|---|---|
class |
CropAndResize |
class |
ExtractImagePatches |
class |
ImageResize |
class |
NonMaxSuppression |
class |
NonMaxSuppressionV3 |
class |
NonMaxSuppressionWithOverlaps |
class |
ResizeArea |
class |
ResizeBicubic |
class |
ResizeBilinear |
class |
ResizeNearestNeighbor |
| Modifier and Type | Class and Description |
|---|---|
class |
ArgAmax |
class |
ArgAmin |
class |
ArgMax |
class |
ArgMin |
| Modifier and Type | Class and Description |
|---|---|
class |
ExternalErrorsFunction |
| Modifier and Type | Class and Description |
|---|---|
class |
GRU |
class |
GRUBp |
class |
GRUCell |
class |
LSTMBlock |
class |
LSTMBlockCell |
class |
LSTMCell |
class |
LSTMLayer |
class |
LSTMLayerBp |
class |
SRU |
class |
SRUCell |
| Modifier and Type | Class and Description |
|---|---|
class |
AbsoluteDifferenceLoss |
class |
BaseLoss |
class |
CosineDistanceLoss |
class |
CtcLoss |
class |
HingeLoss |
class |
HuberLoss |
class |
L2Loss
L2 loss op wrapper
|
class |
LogLoss |
class |
LogPoissonLoss |
class |
MeanPairwiseSquaredErrorLoss
Mean Pairwise Squared Error Loss
|
class |
MeanSquaredErrorLoss |
class |
SigmoidCrossEntropyLoss |
class |
SoftmaxCrossEntropyLoss |
class |
SoftmaxCrossEntropyWithLogitsLoss |
class |
SparseSoftmaxCrossEntropyLossWithLogits |
class |
WeightedCrossEntropyLoss |
| Modifier and Type | Class and Description |
|---|---|
class |
AbsoluteDifferenceLossBp |
class |
BaseLossBp |
class |
CosineDistanceLossBp |
class |
CtcLossBp |
class |
HingeLossBp |
class |
HuberLossBp |
class |
LogLossBp |
class |
LogPoissonLossBp |
class |
MeanPairwiseSquaredErrorLossBp |
class |
MeanSquaredErrorLossBp |
class |
SigmoidCrossEntropyLossBp |
class |
SoftmaxCrossEntropyLossBp |
class |
SoftmaxCrossEntropyWithLogitsLossBp |
class |
SparseSoftmaxCrossEntropyLossWithLogitsBp |
| Modifier and Type | Class and Description |
|---|---|
class |
CbowRound |
class |
SkipGramRound |
| Modifier and Type | Class and Description |
|---|---|
class |
HashCode |
class |
Mmul |
class |
MmulBp |
class |
Moments |
class |
NormalizeMoments |
class |
SufficientStatistics |
class |
TensorMmul
TensorMmul
|
class |
TensorMmulBp |
class |
ZeroFraction |
| Modifier and Type | Class and Description |
|---|---|
class |
BaseReductionBp |
class |
CumProdBp |
class |
CumSumBp |
class |
DotBp |
class |
MaxBp |
class |
MeanBp |
class |
MinBp |
class |
Norm1Bp |
class |
Norm2Bp |
class |
NormMaxBp |
class |
PowBp |
class |
ProdBp |
class |
SquaredNormBp |
class |
StandardDeviationBp |
class |
SumBp |
class |
VarianceBp |
| Modifier and Type | Class and Description |
|---|---|
class |
BatchMmul |
class |
LogSumExp |
| Modifier and Type | Class and Description |
|---|---|
class |
PRelu
Parameterized ReLU op
|
class |
RectifiedLinearDerivative |
| Modifier and Type | Class and Description |
|---|---|
class |
ScatterAdd |
class |
ScatterDiv |
class |
ScatterMax |
class |
ScatterMin |
class |
ScatterMul |
class |
ScatterNd |
class |
ScatterNdAdd |
class |
ScatterNdSub |
class |
ScatterNdUpdate |
class |
ScatterSub |
class |
ScatterUpdate |
| Modifier and Type | Class and Description |
|---|---|
class |
ApplyGradientDescent |
class |
BroadcastDynamicShape |
class |
Concat |
class |
ConfusionMatrix |
class |
Create |
class |
Cross |
class |
Diag |
class |
DiagPart |
class |
ExpandDims |
class |
Eye |
class |
Flatten2D |
class |
Gather
Gather op
|
class |
GatherNd
GatherND op
|
class |
Linspace |
class |
MergeAvg |
class |
MergeMax |
class |
MergeMaxIndex |
class |
MergeSum |
class |
MeshGrid |
class |
OneHot |
class |
OnesLike |
class |
ParallelStack |
class |
Permute |
class |
Rank |
class |
ReductionShape |
class |
Repeat |
class |
Reshape |
class |
SequenceMask |
class |
Shape
Returns the shape of the input array.
|
class |
ShapeN |
class |
Size |
class |
SizeAt |
class |
Slice |
class |
Split
Split op
|
class |
SplitV
SplitV op
|
class |
Squeeze |
class |
Stack |
class |
StridedSlice |
class |
Tile |
class |
Transpose |
class |
Unstack |
class |
ZerosLike |
| Modifier and Type | Class and Description |
|---|---|
class |
ConcatBp |
class |
MergeAvgBp |
class |
MergeMaxBp |
class |
SliceBp |
class |
StridedSliceBp |
class |
TileBp |
| Modifier and Type | Class and Description |
|---|---|
class |
BaseTensorOp |
class |
EmbeddingLookup |
class |
TensorArray |
class |
TensorArrayConcat |
class |
TensorArrayGather |
class |
TensorArrayRead |
class |
TensorArrayScatter |
class |
TensorArraySize |
class |
TensorArraySplit |
class |
TensorArrayWrite |
| Modifier and Type | Class and Description |
|---|---|
class |
Angle |
class |
Assert |
class |
BaseDynamicTransformOp |
class |
BinCount |
class |
CheckNumerics |
class |
Cholesky |
class |
Histogram |
class |
HistogramFixedWidth |
class |
IdentityN |
class |
NthElement |
class |
Pad |
class |
ReluLayer |
| Modifier and Type | Class and Description |
|---|---|
class |
IsMax
[1, 2, 3, 1] -> [0, 0, 1, 0]
|
| Modifier and Type | Class and Description |
|---|---|
class |
ClipByAvgNorm |
class |
ClipByNorm |
class |
ClipByNormBp |
class |
ClipByValue |
| Modifier and Type | Class and Description |
|---|---|
class |
SegmentMax |
class |
SegmentMean |
class |
SegmentMin |
class |
SegmentProd |
class |
SegmentSum |
| Modifier and Type | Class and Description |
|---|---|
class |
Cast |
| Modifier and Type | Class and Description |
|---|---|
class |
CubeBp |
class |
DynamicPartitionBp |
class |
EluBp |
class |
GradientBackwardsMarker |
class |
HardSigmoidBp |
class |
HardTanhBp |
class |
LeakyReLUBp |
class |
LogSoftMaxDerivative |
class |
PReluBp |
class |
RationalTanhBp |
class |
RectifiedTanhBp |
class |
Relu6Derivative |
class |
SeluBp |
class |
SigmoidDerivative |
class |
SoftmaxBp |
class |
SoftPlusBp |
class |
SoftSignBp |
class |
TanhDerivative |
class |
ThresholdReluBp |
| Modifier and Type | Class and Description |
|---|---|
class |
AddOp
Addition operation
|
class |
DivOp
Division operation
|
class |
FloorDivOp |
class |
FloorModOp |
class |
MergeAddOp |
class |
ModOp |
class |
MulOp |
class |
RDivOp
Reverse Division operation
|
class |
RealDivOp
RealDivision operation
|
class |
RSubOp |
class |
SquaredDifferenceOp |
class |
SubOp |
class |
TruncateDivOp |
| Modifier and Type | Class and Description |
|---|---|
class |
AddBpOp |
class |
BaseArithmeticBackpropOp |
class |
DivBpOp |
class |
FloorDivBpOp |
class |
FloorModBpOp |
class |
MergeAddBp |
class |
ModBpOp |
class |
MulBpOp |
class |
RDivBpOp |
class |
RSubBpOp |
class |
SquaredDifferenceBpOp |
class |
SubBpOp |
| Modifier and Type | Class and Description |
|---|---|
class |
Identity |
| Modifier and Type | Class and Description |
|---|---|
class |
UnsortedSegmentMax |
class |
UnsortedSegmentMean |
class |
UnsortedSegmentMin |
class |
UnsortedSegmentProd |
class |
UnsortedSegmentSqrtN |
class |
UnsortedSegmentSum |
| Modifier and Type | Class and Description |
|---|---|
class |
SegmentMaxBp |
class |
SegmentMeanBp |
class |
SegmentMinBp |
class |
SegmentProdBp |
class |
SegmentSumBp |
class |
UnsortedSegmentMaxBp |
class |
UnsortedSegmentMeanBp |
class |
UnsortedSegmentMinBp |
class |
UnsortedSegmentProdBp |
class |
UnsortedSegmentSqrtNBp |
class |
UnsortedSegmentSumBp |
| Modifier and Type | Class and Description |
|---|---|
class |
ELU |
| Modifier and Type | Class and Description |
|---|---|
class |
AdaBeliefUpdater |
class |
AdaDeltaUpdater |
class |
AdaGradUpdater |
class |
AdaMaxUpdater |
class |
AdamUpdater |
class |
AmsGradUpdater |
class |
NadamUpdater |
class |
NesterovsUpdater |
class |
RmsPropUpdater |
class |
SgdUpdater |
| Modifier and Type | Class and Description |
|---|---|
class |
RestoreV2 |
class |
SaveV2 |
| Modifier and Type | Class and Description |
|---|---|
class |
RandomStandardNormal |
| Modifier and Type | Class and Description |
|---|---|
class |
DistributionUniform |
class |
RandomBernoulli |
class |
RandomExponential |
class |
RandomGamma |
class |
RandomNormal |
class |
RandomPoisson |
class |
RandomShuffle |
| Modifier and Type | Class and Description |
|---|---|
class |
Range |
| Modifier and Type | Class and Description |
|---|---|
class |
PrintAffinity |
class |
PrintVariable |
| Modifier and Type | Method and Description |
|---|---|
static INDArray[] |
Nd4j.exec(CustomOp op)
Execute the operation and return the result
|
static INDArray[] |
Nd4j.exec(CustomOp op,
OpContext context)
Execute the operation and return the result
|
| Modifier and Type | Method and Description |
|---|---|
protected String |
OpProfiler.getOpClass(CustomOp op) |
void |
OpProfiler.processOpCall(CustomOp op)
This method tracks op calls
|
void |
OpProfiler.processStackCall(CustomOp op,
long timeStart) |
void |
OpProfiler.timeOpCall(CustomOp op,
long startTime) |
| Modifier and Type | Method and Description |
|---|---|
void |
StringAggregator.putTime(String key,
CustomOp op,
long timeSpent) |
Copyright © 2021. All rights reserved.