Interface CustomOp

    • Method Detail

      • configureFromArguments

        void configureFromArguments()
        This allows a custom op to configure relevant fields from its arguments. This is needed when ops are created via reflection for things like model import.
      • opName

        String opName()
        This method returns op opName as string
        Returns:
      • opHash

        long opHash()
        This method returns LongHash of the opName()
        Returns:
      • isInplaceCall

        boolean isInplaceCall()
        This method returns true if op is supposed to be executed inplace
        Returns:
      • iArgs

        long[] iArgs()
      • tArgs

        double[] tArgs()
      • bArgs

        boolean[] bArgs()
      • addTArgument

        void addTArgument​(double... arg)
      • addIArgument

        void addIArgument​(int... arg)
      • addIArgument

        void addIArgument​(long... arg)
      • addBArgument

        void addBArgument​(boolean... arg)
      • addDArgument

        void addDArgument​(DataType... arg)
      • removeIArgument

        void removeIArgument​(Integer arg)
      • addSArgument

        void addSArgument​(String... args)
      • removeSArgument

        void removeSArgument​(String argument)
      • getSArgument

        String getSArgument​(int index)
      • getBArgument

        Boolean getBArgument​(int index)
      • getIArgument

        Long getIArgument​(int index)
      • numIArguments

        int numIArguments()
      • removeTArgument

        void removeTArgument​(Double arg)
      • getTArgument

        Double getTArgument​(int index)
      • numTArguments

        int numTArguments()
      • numBArguments

        int numBArguments()
      • numDArguments

        int numDArguments()
      • numSArguments

        int numSArguments()
      • addInputArgument

        void addInputArgument​(INDArray... arg)
      • removeInputArgument

        void removeInputArgument​(INDArray arg)
      • getInputArgument

        INDArray getInputArgument​(int index)
      • numInputArguments

        int numInputArguments()
      • addOutputArgument

        void addOutputArgument​(INDArray... arg)
      • removeOutputArgument

        void removeOutputArgument​(INDArray arg)
      • getOutputArgument

        INDArray getOutputArgument​(int index)
      • numOutputArguments

        int numOutputArguments()
      • calculateOutputShape

        List<LongShapeDescriptor> calculateOutputShape()
        Calculate the output shape for this op
        Returns:
        Output array shapes
      • calculateOutputShape

        List<LongShapeDescriptor> calculateOutputShape​(OpContext opContext)
        Calculate the output shape for this op
        Returns:
        Output array shapes
      • getDescriptor

        CustomOpDescriptor getDescriptor()
        Get the custom op descriptor if one is available.
        Returns:
      • assertValidForExecution

        void assertValidForExecution()
        Asserts a valid state for execution, otherwise throws an ND4JIllegalStateException
      • clearArrays

        void clearArrays()
        Clear the input and output INDArrays, if any are set