Package org.platanios.tensorflow.proto

Interface KernelDefOrBuilder

  • All Superinterfaces:
    com.google.protobuf.MessageLiteOrBuilder, com.google.protobuf.MessageOrBuilder
    All Known Implementing Classes:
    KernelDef, KernelDef.Builder
    public interface KernelDefOrBuilder
extends com.google.protobuf.MessageOrBuilder

    • Method Summary

      All Methods Instance Methods Abstract Methods 
      Modifier and Type Method Description
      KernelDef.AttrConstraint getConstraint​(int index)
      repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
      int getConstraintCount()
      repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
      java.util.List<KernelDef.AttrConstraint> getConstraintList()
      repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
      KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder​(int index)
      repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
      java.util.List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
      repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
      java.lang.String getDeviceType()
      Type of device this kernel runs on.
      com.google.protobuf.ByteString getDeviceTypeBytes()
      Type of device this kernel runs on.
      java.lang.String getHostMemoryArg​(int index)
      Names of the Op's input_/output_args that reside in host memory instead of device memory.
      com.google.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.
      java.util.List<java.lang.String> getHostMemoryArgList()
      Names of the Op's input_/output_args that reside in host memory instead of device memory.
      java.lang.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.
      com.google.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.
      java.lang.String getOp()
      Must match the name of an Op.
      com.google.protobuf.ByteString getOpBytes()
      Must match the name of an Op.
      int getPriority()
      Prioritization of kernel amongst different devices.

      • Methods inherited from interface com.google.protobuf.MessageLiteOrBuilder

        isInitialized

      • Methods inherited from interface com.google.protobuf.MessageOrBuilder

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

    • Method Detail

      • getOp

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

      • getOpBytes

        com.google.protobuf.ByteString getOpBytes()
         Must match the name of an Op.
        string op = 1;
        Returns:
        The bytes for op.

      • getDeviceType

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

      • getDeviceTypeBytes

        com.google.protobuf.ByteString getDeviceTypeBytes()
         Type of device this kernel runs on.
        string device_type = 2;
        Returns:
        The bytes for deviceType.

      • getConstraintList

        java.util.List<KernelDef.AttrConstraint> getConstraintList()
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • getConstraint

        KernelDef.AttrConstraint getConstraint​(int index)
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • getConstraintCount

        int getConstraintCount()
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • getConstraintOrBuilderList

        java.util.List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • getConstraintOrBuilder

        KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder​(int index)
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • getHostMemoryArgList

        java.util.List<java.lang.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

        java.lang.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

        com.google.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

        java.lang.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

        com.google.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.

      • getPriority

        int getPriority()
         Prioritization of kernel amongst different devices. By default we assume
 priority is 0. The higher the priority the better. By default (i.e. if
 this is not set), we prefer GPU kernels over CPU.
        int32 priority = 6;
        Returns:
        The priority.