Interface KernelDefOrBuilder

  • All Superinterfaces:
    org.nd4j.shade.protobuf.MessageLiteOrBuilder, org.nd4j.shade.protobuf.MessageOrBuilder
    All Known Implementing Classes:
    KernelDef, KernelDef.Builder

    public interface KernelDefOrBuilder
    extends org.nd4j.shade.protobuf.MessageOrBuilder
    • Method Summary

      All Methods Instance Methods Abstract Methods 
      Modifier and Type Method Description
      KernelDef.AttrConstraint getConstraint​(int index)
      repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      int getConstraintCount()
      repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      List<KernelDef.AttrConstraint> getConstraintList()
      repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder​(int index)
      repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
      repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      String getDeviceType()
      Type of device this kernel runs on.
      org.nd4j.shade.protobuf.ByteString getDeviceTypeBytes()
      Type of device this kernel runs on.
      String getHostMemoryArg​(int index)
      Names of the Op's input_/output_args that reside in host memory instead of device memory.
      org.nd4j.shade.protobuf.ByteString getHostMemoryArgBytes​(int index)
      Names of the Op's input_/output_args that reside in host memory instead of device memory.
      int getHostMemoryArgCount()
      Names of the Op's input_/output_args that reside in host memory instead of device memory.
      List<String> getHostMemoryArgList()
      Names of the Op's input_/output_args that reside in host memory instead of device memory.
      String getLabel()
      This allows experimental kernels to be registered for an op that won't be used unless the user specifies a "_kernel" attr with value matching this.
      org.nd4j.shade.protobuf.ByteString getLabelBytes()
      This allows experimental kernels to be registered for an op that won't be used unless the user specifies a "_kernel" attr with value matching this.
      String getOp()
      Must match the name of an Op.
      org.nd4j.shade.protobuf.ByteString getOpBytes()
      Must match the name of an Op.
      • Methods inherited from interface org.nd4j.shade.protobuf.MessageLiteOrBuilder

        isInitialized
      • Methods inherited from interface org.nd4j.shade.protobuf.MessageOrBuilder

        findInitializationErrors, getAllFields, getDefaultInstanceForType, getDescriptorForType, getField, getInitializationErrorString, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof
    • Method Detail

      • getOp

        String getOp()
         Must match the name of an Op.
         
        string op = 1;
        Returns:
        The op.
      • getOpBytes

        org.nd4j.shade.protobuf.ByteString getOpBytes()
         Must match the name of an Op.
         
        string op = 1;
        Returns:
        The bytes for op.
      • getDeviceType

        String getDeviceType()
         Type of device this kernel runs on.
         
        string device_type = 2;
        Returns:
        The deviceType.
      • getDeviceTypeBytes

        org.nd4j.shade.protobuf.ByteString getDeviceTypeBytes()
         Type of device this kernel runs on.
         
        string device_type = 2;
        Returns:
        The bytes for deviceType.
      • getConstraint

        KernelDef.AttrConstraint getConstraint​(int index)
        repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      • getConstraintCount

        int getConstraintCount()
        repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      • getHostMemoryArgList

        List<String> getHostMemoryArgList()
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Returns:
        A list containing the hostMemoryArg.
      • getHostMemoryArgCount

        int getHostMemoryArgCount()
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Returns:
        The count of hostMemoryArg.
      • getHostMemoryArg

        String getHostMemoryArg​(int index)
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Parameters:
        index - The index of the element to return.
        Returns:
        The hostMemoryArg at the given index.
      • getHostMemoryArgBytes

        org.nd4j.shade.protobuf.ByteString getHostMemoryArgBytes​(int index)
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Parameters:
        index - The index of the value to return.
        Returns:
        The bytes of the hostMemoryArg at the given index.
      • getLabel

        String getLabel()
         This allows experimental kernels to be registered for an op that
         won't be used unless the user specifies a "_kernel" attr with
         value matching this.
         
        string label = 5;
        Returns:
        The label.
      • getLabelBytes

        org.nd4j.shade.protobuf.ByteString getLabelBytes()
         This allows experimental kernels to be registered for an op that
         won't be used unless the user specifies a "_kernel" attr with
         value matching this.
         
        string label = 5;
        Returns:
        The bytes for label.