Uses of Interface
org.nd4j.linalg.api.ops.CustomOp
-
-
Uses of CustomOp in org.nd4j.linalg.api.ops
Classes in org.nd4j.linalg.api.ops that implement CustomOp Modifier and Type Class Description classDynamicCustomOpclassNoOpMethods in org.nd4j.linalg.api.ops that return CustomOp Modifier and Type Method Description CustomOpBaseOp. toCustomOp()CustomOpOp. toCustomOp() -
Uses of CustomOp in org.nd4j.linalg.api.ops.compat
Classes in org.nd4j.linalg.api.ops.compat that implement CustomOp Modifier and Type Class Description classCompatSparseToDenseclassCompatStringSplit -
Uses of CustomOp in org.nd4j.linalg.api.ops.compression
Classes in org.nd4j.linalg.api.ops.compression that implement CustomOp Modifier and Type Class Description classDecodeBitmapclassDecodeThresholdclassEncodeBitmapclassEncodeThreshold -
Uses of CustomOp in org.nd4j.linalg.api.ops.custom
Classes in org.nd4j.linalg.api.ops.custom that implement CustomOp Modifier and Type Class Description classAdjustContrastclassAdjustHueclassAdjustSaturationclassBarnesEdgeForcesclassBarnesHutGainsclassBarnesHutSymmetrizeclassBetaIncclassBitCastclassCompareAndBitpackclassDigammaclassDivideNoNanclassDrawBoundingBoxesclassEigclassFakeQuantWithMinMaxVarsPerChannelclassFlattenclassFusedBatchNormclassHsvToRgbclassIgammaclassIgammacclassInvokeInvoke is an opclassKnnMinDistanceclassLgammaclassLinearSolveclassLogdetclassLstsqclassLuclassMatrixBandPartclassPolygammaclassRandomCropclassRgbToGrayscaleclassRgbToHsvclassRgbToYiqclassRgbToYuvclassRollclassSpTreeCellclassToggleBitsclassTriclassTriangularSolveclassTriuclassTriuBpclassYiqToRgbclassYuvToRgb -
Uses of CustomOp in org.nd4j.linalg.api.ops.executioner
Methods in org.nd4j.linalg.api.ops.executioner that return CustomOp Modifier and Type Method Description CustomOpDefaultOpExecutioner. execAndReturn(CustomOp op)CustomOpOpExecutioner. execAndReturn(CustomOp op)This method executes given CustomOp PLEASE NOTE: You're responsible for input/output validationMethods in org.nd4j.linalg.api.ops.executioner with parameters of type CustomOp Modifier and Type Method Description INDArray[]DefaultOpExecutioner. allocateOutputArrays(CustomOp op)INDArray[]OpExecutioner. allocateOutputArrays(CustomOp op)Equivalent to calliList<LongShapeDescriptor>DefaultOpExecutioner. calculateOutputShape(CustomOp op)List<LongShapeDescriptor>DefaultOpExecutioner. calculateOutputShape(CustomOp op, OpContext opContext)List<LongShapeDescriptor>OpExecutioner. calculateOutputShape(CustomOp op)List<LongShapeDescriptor>OpExecutioner. calculateOutputShape(CustomOp op, OpContext opContext)static voidOpExecutionerUtil. checkForInf(CustomOp op, OpContext oc)static voidOpExecutionerUtil. checkForNaN(CustomOp op, OpContext oc)protected voidDefaultOpExecutioner. checkForWorkspaces(CustomOp op, OpContext oc)INDArray[]DefaultOpExecutioner. exec(CustomOp op)INDArray[]DefaultOpExecutioner. exec(CustomOp op, OpContext context)INDArray[]OpExecutioner. exec(CustomOp op)INDArray[]OpExecutioner. exec(CustomOp op, OpContext context)This method executes op with given contextCustomOpDefaultOpExecutioner. execAndReturn(CustomOp op)CustomOpOpExecutioner. execAndReturn(CustomOp op)This method executes given CustomOp PLEASE NOTE: You're responsible for input/output validationlongDefaultOpExecutioner. profilingConfigurableHookIn(CustomOp op, OpContext oc)voidDefaultOpExecutioner. profilingConfigurableHookOut(CustomOp op, OpContext oc, long timeStart)longDefaultOpExecutioner. profilingHookIn(CustomOp op, OpContext oc)Deprecated.voidDefaultOpExecutioner. profilingHookOut(CustomOp op, OpContext oc, long timeStart)Deprecated. -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.broadcast
Classes in org.nd4j.linalg.api.ops.impl.broadcast that implement CustomOp Modifier and Type Class Description classBiasAddBias addition gradient operation.classBiasAddGradclassBroadcastTo -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.controlflow
Classes in org.nd4j.linalg.api.ops.impl.controlflow that implement CustomOp Modifier and Type Class Description classSelectclassWhereclassWhereNumpy -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.controlflow.compat
Classes in org.nd4j.linalg.api.ops.impl.controlflow.compat that implement CustomOp Modifier and Type Class Description classBaseCompatOpclassEnterclassExitclassLoopCondclassMergeclassNextIterationclassStopGradientclassSwitchclassWhile -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.image
Classes in org.nd4j.linalg.api.ops.impl.image that implement CustomOp Modifier and Type Class Description classCropAndResizeclassExtractImagePatchesclassImageResizeclassNonMaxSuppressionclassNonMaxSuppressionV3classNonMaxSuppressionWithOverlapsclassResizeAreaclassResizeBicubicclassResizeBilinearclassResizeNearestNeighbor -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.indexaccum.custom
Classes in org.nd4j.linalg.api.ops.impl.indexaccum.custom that implement CustomOp Modifier and Type Class Description classArgAmaxclassArgAminclassArgMaxclassArgMin -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.layers
Classes in org.nd4j.linalg.api.ops.impl.layers that implement CustomOp Modifier and Type Class Description classExternalErrorsFunction -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.layers.convolution
-
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.layers.recurrent
Classes in org.nd4j.linalg.api.ops.impl.layers.recurrent that implement CustomOp Modifier and Type Class Description classGRUclassGRUBpclassGRUCellclassLSTMBlockclassLSTMBlockCellclassLSTMCellclassLSTMLayerclassLSTMLayerBpclassSRUclassSRUCell -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.loss
Classes in org.nd4j.linalg.api.ops.impl.loss that implement CustomOp Modifier and Type Class Description classAbsoluteDifferenceLossclassBaseLossclassCosineDistanceLossclassCtcLossclassHingeLossclassHuberLossclassL2LossL2 loss op wrapperclassLogLossclassLogPoissonLossclassMeanPairwiseSquaredErrorLossMean Pairwise Squared Error LossclassMeanSquaredErrorLossclassSigmoidCrossEntropyLossclassSoftmaxCrossEntropyLossclassSoftmaxCrossEntropyWithLogitsLossclassSparseSoftmaxCrossEntropyLossWithLogitsclassWeightedCrossEntropyLoss -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.loss.bp
Classes in org.nd4j.linalg.api.ops.impl.loss.bp that implement CustomOp Modifier and Type Class Description classAbsoluteDifferenceLossBpclassBaseLossBpclassCosineDistanceLossBpclassCtcLossBpclassHingeLossBpclassHuberLossBpclassLogLossBpclassLogPoissonLossBpclassMeanPairwiseSquaredErrorLossBpclassMeanSquaredErrorLossBpclassSigmoidCrossEntropyLossBpclassSoftmaxCrossEntropyLossBpclassSoftmaxCrossEntropyWithLogitsLossBpclassSparseSoftmaxCrossEntropyLossWithLogitsBp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.nlp
Classes in org.nd4j.linalg.api.ops.impl.nlp that implement CustomOp Modifier and Type Class Description classCbowRoundclassSkipGramRound -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.reduce
Classes in org.nd4j.linalg.api.ops.impl.reduce that implement CustomOp Modifier and Type Class Description classHashCodeclassMmulclassMmulBpclassMomentsclassNormalizeMomentsclassSufficientStatisticsclassTensorMmulTensorMmulclassTensorMmulBpclassZeroFraction -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.reduce.bp
Classes in org.nd4j.linalg.api.ops.impl.reduce.bp that implement CustomOp Modifier and Type Class Description classBaseReductionBpclassCumProdBpclassCumSumBpclassDotBpclassMaxBpclassMeanBpclassMinBpclassNorm1BpclassNorm2BpclassNormMaxBpclassPowBpclassProdBpclassSquaredNormBpclassStandardDeviationBpclassSumBpclassVarianceBp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.reduce.custom
Classes in org.nd4j.linalg.api.ops.impl.reduce.custom that implement CustomOp Modifier and Type Class Description classBaseDynamicCustomBoolReductionclassBaseDynamicCustomIndexReductionclassBaseDynamicCustomLongReductionclassBaseDynamicCustomReductionBase class for reduction.classBatchMmulclassLogSumExp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.scalar
Classes in org.nd4j.linalg.api.ops.impl.scalar that implement CustomOp Modifier and Type Class Description classPReluParameterized ReLU opclassRectifiedLinearDerivative -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.scatter
Classes in org.nd4j.linalg.api.ops.impl.scatter that implement CustomOp Modifier and Type Class Description classScatterAddclassScatterDivclassScatterMaxclassScatterMinclassScatterMulclassScatterNdclassScatterNdAddclassScatterNdSubclassScatterNdUpdateclassScatterSubclassScatterUpdate -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.shape
Classes in org.nd4j.linalg.api.ops.impl.shape that implement CustomOp Modifier and Type Class Description classApplyGradientDescentclassBroadcastDynamicShapeclassConcatclassConfusionMatrixclassCreateclassCreateViewclassCrossclassDiagclassDiagPartclassExpandDimsclassEyeclassFlatten2DclassGatherGather opclassGatherNdGatherND opclassLinspaceclassMergeAvgclassMergeMaxclassMergeMaxIndexclassMergeSumclassMeshGridclassOneHotclassOnesAsclassOnesLikeclassParallelStackclassPermuteclassRankclassReductionShapeclassRepeatclassReshapeclassSequenceMaskclassSetShapeReturns the shape of the input array.classShapeReturns the shape of the input array.classShapeNclassSizeclassSizeAtclassSliceclassSplitSplit opclassSplitVSplitV opclassSqueezeclassStackclassStridedSliceclassTileclassTransposeclassUnstackclassZerosLike -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.shape.bp
Classes in org.nd4j.linalg.api.ops.impl.shape.bp that implement CustomOp Modifier and Type Class Description classConcatBpclassMergeAvgBpclassMergeMaxBpclassSliceBpclassStridedSliceBpclassTileBp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.shape.tensorops
Classes in org.nd4j.linalg.api.ops.impl.shape.tensorops that implement CustomOp Modifier and Type Class Description classBaseTensorOpclassEmbeddingLookupclassTensorArrayclassTensorArrayConcatclassTensorArrayGatherclassTensorArrayReadclassTensorArrayRemoveclassTensorArrayScatterclassTensorArraySizeclassTensorArraySplitclassTensorArrayWrite -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms
Classes in org.nd4j.linalg.api.ops.impl.transforms that implement CustomOp Modifier and Type Class Description classAngleclassAssertclassBaseDynamicTransformOpclassBinCountclassCheckNumericsclassCholeskyclassHistogramclassHistogramFixedWidthclassIdentityNclassNthElementclassPadclassReluLayer -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.any
Classes in org.nd4j.linalg.api.ops.impl.transforms.any that implement CustomOp Modifier and Type Class Description classIsMax[1, 2, 3, 1] -> [0, 0, 1, 0] -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.clip
Classes in org.nd4j.linalg.api.ops.impl.transforms.clip that implement CustomOp Modifier and Type Class Description classClipByAvgNormclassClipByNormclassClipByNormBpclassClipByValue -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.custom
-
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.custom.segment
Classes in org.nd4j.linalg.api.ops.impl.transforms.custom.segment that implement CustomOp Modifier and Type Class Description classSegmentMaxclassSegmentMeanclassSegmentMinclassSegmentProdclassSegmentSum -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.dtype
Classes in org.nd4j.linalg.api.ops.impl.transforms.dtype that implement CustomOp Modifier and Type Class Description classCastclassMinMaxDataTypeReturns the min or max (0 or 1 in int arguments) for a given data type. -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.floating
Classes in org.nd4j.linalg.api.ops.impl.transforms.floating that implement CustomOp Modifier and Type Class Description classSqrtM -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.gradient
Classes in org.nd4j.linalg.api.ops.impl.transforms.gradient that implement CustomOp Modifier and Type Class Description classCubeBpclassDynamicPartitionBpclassEluBpclassGradientBackwardsMarkerclassHardSigmoidBpclassHardTanhBpclassLeakyReLUBpclassLogSoftMaxDerivativeclassPReluBpclassRationalTanhBpclassRectifiedTanhBpclassRelu6DerivativeclassSeluBpclassSigmoidDerivativeclassSoftmaxBpclassSoftPlusBpclassSoftSignBpclassTanhDerivativeclassThresholdReluBp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.pairwise.arithmetic
Classes in org.nd4j.linalg.api.ops.impl.transforms.pairwise.arithmetic that implement CustomOp Modifier and Type Class Description classAddOpAddition operationclassDivOpDivision operationclassFloorDivOpclassFloorModOpclassMergeAddOpclassModOpclassMulOpclassRDivOpReverse Division operationclassRealDivOpRealDivision operationclassRSubOpclassSquaredDifferenceOpclassSubOpclassTruncateDivOp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.pairwise.arithmetic.bp
Classes in org.nd4j.linalg.api.ops.impl.transforms.pairwise.arithmetic.bp that implement CustomOp Modifier and Type Class Description classAddBpOpclassBaseArithmeticBackpropOpclassDivBpOpclassFloorDivBpOpclassFloorModBpOpclassMergeAddBpclassModBpOpclassMulBpOpclassRDivBpOpclassRSubBpOpclassSquaredDifferenceBpOpclassSubBpOp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.same
Classes in org.nd4j.linalg.api.ops.impl.transforms.same that implement CustomOp Modifier and Type Class Description classIdentity -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.segment
Classes in org.nd4j.linalg.api.ops.impl.transforms.segment that implement CustomOp Modifier and Type Class Description classUnsortedSegmentMaxclassUnsortedSegmentMeanclassUnsortedSegmentMinclassUnsortedSegmentProdclassUnsortedSegmentSqrtNclassUnsortedSegmentSum -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.segment.bp
Classes in org.nd4j.linalg.api.ops.impl.transforms.segment.bp that implement CustomOp Modifier and Type Class Description classSegmentMaxBpclassSegmentMeanBpclassSegmentMinBpclassSegmentProdBpclassSegmentSumBpclassUnsortedSegmentMaxBpclassUnsortedSegmentMeanBpclassUnsortedSegmentMinBpclassUnsortedSegmentProdBpclassUnsortedSegmentSqrtNBpclassUnsortedSegmentSumBp -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.transforms.strict
Classes in org.nd4j.linalg.api.ops.impl.transforms.strict that implement CustomOp Modifier and Type Class Description classELU -
Uses of CustomOp in org.nd4j.linalg.api.ops.impl.updaters
Classes in org.nd4j.linalg.api.ops.impl.updaters that implement CustomOp Modifier and Type Class Description classAdaBeliefUpdaterclassAdaDeltaUpdaterclassAdaGradUpdaterclassAdaMaxUpdaterclassAdamUpdaterclassAmsGradUpdaterclassNadamUpdaterclassNesterovsUpdaterclassRmsPropUpdaterclassSgdUpdater -
Uses of CustomOp in org.nd4j.linalg.api.ops.persistence
Classes in org.nd4j.linalg.api.ops.persistence that implement CustomOp Modifier and Type Class Description classRestoreV2classSaveV2 -
Uses of CustomOp in org.nd4j.linalg.api.ops.random.compat
Classes in org.nd4j.linalg.api.ops.random.compat that implement CustomOp Modifier and Type Class Description classRandomStandardNormal -
Uses of CustomOp in org.nd4j.linalg.api.ops.random.custom
Classes in org.nd4j.linalg.api.ops.random.custom that implement CustomOp Modifier and Type Class Description classDistributionUniformclassRandomBernoulliclassRandomExponentialclassRandomGammaclassRandomNormalclassRandomPoissonclassRandomShuffle -
Uses of CustomOp in org.nd4j.linalg.api.ops.random.impl
Classes in org.nd4j.linalg.api.ops.random.impl that implement CustomOp Modifier and Type Class Description classRandomMultinomialclassRange -
Uses of CustomOp in org.nd4j.linalg.api.ops.util
Classes in org.nd4j.linalg.api.ops.util that implement CustomOp Modifier and Type Class Description classPrintAffinityclassPrintVariable -
Uses of CustomOp in org.nd4j.linalg.factory
Methods in org.nd4j.linalg.factory with parameters of type CustomOp Modifier and Type Method Description static INDArray[]Nd4j. exec(CustomOp op)Execute the operation and return the resultstatic INDArray[]Nd4j. exec(CustomOp op, OpContext context)Execute the operation and return the result -
Uses of CustomOp in org.nd4j.linalg.profiler
Methods in org.nd4j.linalg.profiler with parameters of type CustomOp Modifier and Type Method Description protected StringOpProfiler. getOpClass(CustomOp op)voidOpProfiler. processOpCall(CustomOp op)This method tracks op callsvoidOpProfiler. processStackCall(CustomOp op, long timeStart)voidOpProfiler. timeOpCall(CustomOp op, long startTime) -
Uses of CustomOp in org.nd4j.linalg.profiler.data
Methods in org.nd4j.linalg.profiler.data with parameters of type CustomOp Modifier and Type Method Description voidStringAggregator. putTime(String key, CustomOp op, long timeSpent) -
Uses of CustomOp in org.nd4j.linalg.util
Methods in org.nd4j.linalg.util with parameters of type CustomOp Modifier and Type Method Description static voidLinAlgExceptions. assertAllConfigured(CustomOp op, int numExpectedArgs)Asserts that at least the number of arguments on the given op is met.
-