Modifier and Type | Method and Description |
---|---|
List<int[]> |
SameDiffOpExecutioner.calculateOutputShape(CustomOp op) |
void |
SameDiffOpExecutioner.exec(CustomOp op)
This method executes given CustomOp
|
Modifier and Type | Interface and Description |
---|---|
interface |
Module
A Module is a
CustomOp
with varying input arguments
and automatically calculated outputs
defined at a higher level than c++. |
Modifier and Type | Class and Description |
---|---|
class |
BaseModule
Abstract base class for
Module
that handles Dynamic ops and handles nesting. |
class |
DynamicCustomOp
Basic implementation for CustomOp
|
class |
NoOp |
Modifier and Type | Method and Description |
---|---|
CustomOp |
Op.toCustomOp()
|
CustomOp |
BaseOp.toCustomOp() |
Modifier and Type | Class and Description |
---|---|
class |
ScatterUpdate |
Modifier and Type | Field and Description |
---|---|
protected CustomOp |
ScatterUpdate.op |
Modifier and Type | Method and Description |
---|---|
List<int[]> |
DefaultOpExecutioner.calculateOutputShape(CustomOp op) |
List<int[]> |
OpExecutioner.calculateOutputShape(CustomOp op) |
void |
DefaultOpExecutioner.exec(CustomOp op) |
void |
OpExecutioner.exec(CustomOp op)
This method executes given CustomOp
PLEASE NOTE: You're responsible for input/output validation
|
Modifier and Type | Class and Description |
---|---|
class |
CumProd |
class |
CumSum
Cumulative sum operation, optionally along dimension.
|
class |
Mmul
Matrix multiplication/dot product
|
class |
Moments |
class |
NormalizeMoments |
class |
SigmoidCrossEntropyLoss
Sigmoid cross entropy loss with logits
|
class |
SoftmaxCrossEntropyLoss
Softmax cross entropy loss with logits
|
class |
TensorMmul
TensorMmul
|
class |
WeightedCrossEntropyLoss
Weighted cross entropy loss with logits
|
class |
ZeroFraction
Compute the fraction of zero elements
|
Modifier and Type | Class and Description |
---|---|
class |
BiasAdd |
Modifier and Type | Class and Description |
---|---|
class |
If
Equivalent to tensorflow's conditional op.
|
class |
IfDerivative |
class |
Select |
class |
Where |
class |
WhereNumpy |
class |
While
Equivalent to tensorflow's while loop
Takes in:
loopVars
loop body
condition
runs loop till condition is false.
|
class |
WhileDerivative
While loop derivative
|
Modifier and Type | Class and Description |
---|---|
class |
BaseCompatOp |
class |
Enter |
class |
Exit |
class |
LoopCond |
class |
Merge |
class |
NextIteration |
class |
Switch |
Modifier and Type | Class and Description |
---|---|
class |
Linear
Linear:
a * bT
|
Modifier and Type | Class and Description |
---|---|
class |
AvgPooling2D
Average Pooling2D operation
|
class |
BatchNorm
BatchNorm operation
|
class |
BatchNormDerivative
BatchNormDerivative operation
|
class |
Col2Im
Col2Im operation.
|
class |
Conv1D
Conv2D operation
|
class |
Conv2D
Conv2D operation
|
class |
Conv2DDerivative
Conv2DDerivative operation
|
class |
Conv3D
Conv3D operation
|
class |
Conv3DDerivative
Conv3DDerivative operation
|
class |
DeConv2D
DeConv2D operation
|
class |
DeConv2DDerivative
DeConv2DDerivative operation
|
class |
DepthToSpace
Inverse operation to SpaceToDepth.
|
class |
DepthwiseConv2D
Depthwise Conv2D operation
|
class |
FullConv3D
FullConv3D operation
|
class |
FullConv3DDerivative
FullConv3DDerivative operation
|
class |
Im2col
Im2col operation
|
class |
LocalResponseNormalization
LocalResponseNormalization operation
|
class |
LocalResponseNormalizationDerivative
LocalResponseNormalizationDerivative operation
|
class |
MaxPooling2D
Max Pooling2D operation
|
class |
Pooling2D
Pooling2D operation
|
class |
Pooling2DDerivative
Pooling2DDerivative operation
|
class |
Pooling3D
Pooling3D operation
|
class |
Pooling3DDerivative
Pooling3DDerivative operation
|
class |
SConv2D
Separable convolution 2D operation
|
class |
SConv2DDerivative
SConv2DDerivative operation
|
class |
SpaceToDepth
This operation takes 4D array in, in either NCHW or NHWC format, and moves data from spatial dimensions (HW)
to channels (C) for given blockSize
|
class |
Upsampling
Upsampling operation
|
class |
UpsamplingDerivative
UpsamplingDerivative operation
|
Modifier and Type | Class and Description |
---|---|
class |
GRUCell
GRU cell for RNNs
|
class |
LSTMCell
LSTM cell
|
class |
SRU
Simple recurrent unit
|
class |
SRUCell
A simple recurrent unit cell.
|
Modifier and Type | Class and Description |
---|---|
class |
ScatterAdd
Created by farizrahman4u on 3/23/18.
|
class |
ScatterDiv
Created by farizrahman4u on 3/23/18.
|
class |
ScatterMul
Created by farizrahman4u on 3/23/18.
|
class |
ScatterSub
Created by farizrahman4u on 3/23/18.
|
Modifier and Type | Class and Description |
---|---|
class |
ApplyGradientDescent
Reshape function
|
class |
Broadcast
Broadcast function
|
class |
Concat |
class |
ConfusionMatrix |
class |
Cross
Pairwise cross-product of two tensors of the same shape.
|
class |
Diag
Computes a diagonal matrix of shape (n, n) from a vector of length n.
|
class |
DiagPart
Return the diagonal part of a tensor.
|
class |
ExpandDims
ExpandDims function
|
class |
Eye
Computes a batch of identity matrices of shape (numRows, numCols), returns a single tensor.
|
class |
Gather
Gather op
|
class |
GatherNd
GatherND op
|
class |
MergeAvg |
class |
MergeMax |
class |
MergeSum |
class |
OneHot
Created by susaneraly on 3/14/18.
|
class |
OnesLike
OnesLike function - gives an output array with all values/entries being 1, with the same shape as the input.
|
class |
ParallelStack
Stacks n input tensors of same shape to tensor of rank n + 1.
|
class |
Permute
Permute function
|
class |
Rank
Rank function
|
class |
Repeat
Repeat function
|
class |
Reshape
Reshape function
|
class |
SequenceMask
Created by farizrahman4u on 3/28/18.
|
class |
Shape
Returns the shape of the input array.
|
class |
Slice
Slice function
|
class |
Split
SplitV op
|
class |
SplitV
SplitV op
|
class |
Squeeze |
class |
Stack
Stack operation.
|
class |
StridedSlice
Strided Slice function
|
class |
Tile
Tile function
|
class |
Transpose
Transpose function
|
class |
Unstack
Unstack op conversion
|
class |
ZerosLike
Reshape function
|
Modifier and Type | Class and Description |
---|---|
class |
BaseTensorOp |
class |
TensorArrayGatherV3 |
class |
TensorArrayReadV3 |
class |
TensorArrayScatterV3 |
class |
TensorArrayV3 |
class |
TensorArrayWriteV3 |
class |
TensorSizeV3 |
Modifier and Type | Class and Description |
---|---|
class |
Assign
Assign op: x = y, with broadcast as required
|
class |
ATan2
Arc Tangent elementwise function
|
class |
BaseDynamicTransformOp |
class |
BatchToSpace
N-dimensional batch to space operation.
|
class |
Dilation2D
Dilation2D op wrapper
|
class |
DynamicPartition
Transforms a given input tensor into numPartitions partitions, as indicated by the indices in "partitions".
|
class |
DynamicStitch
Transforms a given input tensor into numPartitions partitions, as indicated by the indices in "partitions".
|
class |
Fill
Fill an array of given "shape" with the provided "value", e.g.
|
class |
Identity
Identity function
|
class |
InvertPermutation
Inverse of index permutation.
|
class |
MatrixDiagPart |
class |
MatrixSetDiag |
class |
ReluLayer
Composed op: relu((X, W) + b)
|
class |
Reverse |
class |
ReverseSequence
Created by farizrahman4u on 3/16/18.
|
class |
SoftMax
Soft max function
row_maxes is a row vector (max for each row)
row_maxes = rowmaxes(input)
diff = exp(input - max) / diff.rowSums()
Outputs a probability distribution.
|
class |
SpaceToBatch
N-dimensional space to batch operation.
|
class |
XwPlusB
Composed op: mmul (X, W) + b
|
Modifier and Type | Class and Description |
---|---|
class |
AddOp
Addition operation
|
class |
DivOp
Division operation
|
class |
FloorDivOp
Truncated division operation
|
class |
MergeAddOp
Addition operation for n operands, called "mergeadd" in libnd4j
|
class |
MulOp
Multiplication operation
|
class |
RDivOp
OldReverse Division operation
|
class |
RealDivOp
RealDivision operation
|
class |
RSubOp
OldReverse 'Subtraction operation
|
class |
SquaredDifferenceOp
Squared difference operation, i.e.
|
class |
SubOp
Subtraction operation
|
Modifier and Type | Class and Description |
---|---|
class |
AddBpOp
Addition backprop operation.
|
class |
BaseArithmeticBackpropOp
Base arithmetic backprop operation
|
class |
DivBpOp
Division backprop operation.
|
class |
FloorDivBpOp
Floor div backprop operation.
|
class |
FloorModBpOp
Floor div backprop operation.
|
class |
MulBpOp
Division backprop operation.
|
class |
RDivBpOp
Division backprop operation.
|
class |
RSubBpOp
Division backprop operation.
|
class |
SubBpOp
Division backprop operation.
|
Modifier and Type | Class and Description |
---|---|
class |
ClipByNorm |
class |
ClipByValue |
Modifier and Type | Class and Description |
---|---|
class |
Choose
This op allows us to (based on the passed in condition)
to return the element fulfilling the condition.
|
class |
EqualTo
Bit mask over the ndarrays as to whether
the components are equal or not
|
class |
GreaterThan
Bit mask over the ndarrays as to whether
the components are greater than or not
|
class |
GreaterThanOrEqual
Bit mask over the ndarrays as to whether
the components are greater than or equal or not
|
class |
IsNonDecreasing
This op takes 1 n-dimensional array as input,
and returns true if for every adjacent pair we have x[i] <= x[i+1].
|
class |
IsNumericTensor
This op takes 1 n-dimensional array as input, and returns true if input is a numeric array.
|
class |
IsStrictlyIncreasing
This op takes 1 n-dimensional array as input,
and returns true if for every adjacent pair we have x[i] < x[i+1].
|
class |
LessThan
Bit mask over the ndarrays as to whether
the components are less than or not
|
class |
LessThanOrEqual
Bit mask over the ndarrays as to whether
the components are less than or equal or not
|
class |
ListDiff |
class |
Max
Max function
|
class |
Min
Min function
|
class |
NotEqualTo
Not equal to function:
Bit mask over whether 2 elements are not equal or not
|
Modifier and Type | Class and Description |
---|---|
class |
Cast
Cast op wrapper.
|
Modifier and Type | Class and Description |
---|---|
class |
RestoreV2 |
class |
SaveV2 |
Modifier and Type | Class and Description |
---|---|
class |
DistributionUniform
Uniform distribution wrapper
|
Modifier and Type | Class and Description |
---|---|
class |
Range
Range Op implementation, generates from..to distribution within Z
|
Copyright © 2018. All rights reserved.