org.tensorflow.framework.KernelDefOrBuilder Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation
The newest version!
// Generated by the protocol buffer compiler.  DO NOT EDIT!
// source: tensorflow/core/framework/kernel_def.proto

package org.tensorflow.framework;

public interface KernelDefOrBuilder extends
    // @@protoc_insertion_point(interface_extends:tensorflow.KernelDef)
    org.nd4j.shade.protobuf.MessageOrBuilder {

  /**
   *    * Must match the name of an Op.
   * 
   *
   * string op = 1;
   * @return The op.
   */
  java.lang.String getOp();
  /**
   *    * Must match the name of an Op.
   * 
   *
   * string op = 1;
   * @return The bytes for op.
   */
  org.nd4j.shade.protobuf.ByteString
      getOpBytes();

  /**
   *    * Type of device this kernel runs on.
   * 
   *
   * string device_type = 2;
   * @return The deviceType.
   */
  java.lang.String getDeviceType();
  /**
   *    * Type of device this kernel runs on.
   * 
   *
   * string device_type = 2;
   * @return The bytes for deviceType.
   */
  org.nd4j.shade.protobuf.ByteString
      getDeviceTypeBytes();

  /**
   * repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
   */
  java.util.List 
      getConstraintList();
  /**
   * repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
   */
  org.tensorflow.framework.KernelDef.AttrConstraint getConstraint(int index);
  /**
   * repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
   */
  int getConstraintCount();
  /**
   * repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
   */
  java.util.List 
      getConstraintOrBuilderList();
  /**
   * repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
   */
  org.tensorflow.framework.KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(
      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;
   * @return A list containing the hostMemoryArg.
   */
  java.util.List
      getHostMemoryArgList();
  /**
   *    * Names of the Op's input_/output_args that reside in host memory
   * instead of device memory.
   * 
   *
   * repeated string host_memory_arg = 4;
   * @return The count of hostMemoryArg.
   */
  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;
   * @param index The index of the element to return.
   * @return The hostMemoryArg at the given index.
   */
  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;
   * @param index The index of the value to return.
   * @return The bytes of the hostMemoryArg at the given index.
   */
  org.nd4j.shade.protobuf.ByteString
      getHostMemoryArgBytes(int index);

  /**
   *    * 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;
   * @return The label.
   */
  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;
   * @return The bytes for label.
   */
  org.nd4j.shade.protobuf.ByteString
      getLabelBytes();
}