Packages

package kernel_def

Ordering
  1. Alphabetic
Visibility
  1. Public
  2. All

Type Members

  1. final case class KernelDef(op: String = "", deviceType: String = "", constraint: Seq[AttrConstraint] = _root_.scala.Seq.empty, hostMemoryArg: Seq[String] = _root_.scala.Seq.empty, label: String = "", priority: Int = 0, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[KernelDef] with Product with Serializable

    op

    Must match the name of an Op.

    deviceType

    Type of device this kernel runs on.

    hostMemoryArg

    Names of the Op's input_/output_args that reside in host memory instead of device memory.

    label

    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.

    priority

    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.

    Annotations
    @SerialVersionUID()
  2. final case class KernelList(kernel: Seq[KernelDef] = _root_.scala.Seq.empty, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[KernelList] with Product with Serializable

    A collection of KernelDefs

    A collection of KernelDefs

    Annotations
    @SerialVersionUID()

Value Members

  1. object KernelDef extends GeneratedMessageCompanion[KernelDef]
  2. object KernelDefProto extends GeneratedFileObject
  3. object KernelList extends GeneratedMessageCompanion[KernelList]

Ungrouped