Interface KernelDefOrBuilder

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

  • Method Details

    • getOp

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

    • getOpBytes

      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

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

    • getConstraintList

      List<KernelDef.AttrConstraint> getConstraintList()
      repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;

    • getConstraint

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

    • getConstraintCount

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

    • getConstraintOrBuilderList

      List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
      repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;

    • getConstraintOrBuilder

      KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(int index)
      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

      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

      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.