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()
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- KernelDef
- Updatable
- GeneratedMessage
- Serializable
- Serializable
- Product
- Equals
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Instance Constructors
-
new
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 = ...)
- 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.
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
- def addAllConstraint(__vs: Iterable[AttrConstraint]): KernelDef
- def addAllHostMemoryArg(__vs: Iterable[String]): KernelDef
- def addConstraint(__vs: AttrConstraint*): KernelDef
- def addHostMemoryArg(__vs: String*): KernelDef
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
- def clearConstraint: KernelDef
- def clearHostMemoryArg: KernelDef
-
def
clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native() @HotSpotIntrinsicCandidate()
-
def
companion: KernelDef.type
- Definition Classes
- KernelDef → GeneratedMessage
- val constraint: Seq[AttrConstraint]
- val deviceType: String
- def discardUnknownFields: KernelDef
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native() @HotSpotIntrinsicCandidate()
-
def
getField(__field: FieldDescriptor): PValue
- Definition Classes
- KernelDef → GeneratedMessage
-
def
getFieldByNumber(__fieldNumber: Int): Any
- Definition Classes
- KernelDef → GeneratedMessage
- val hostMemoryArg: Seq[String]
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- val label: String
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @HotSpotIntrinsicCandidate()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @HotSpotIntrinsicCandidate()
- val op: String
- val priority: Int
-
def
serializedSize: Int
- Definition Classes
- KernelDef → GeneratedMessage
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
final
def
toByteArray: Array[Byte]
- Definition Classes
- GeneratedMessage
-
final
def
toByteString: ByteString
- Definition Classes
- GeneratedMessage
-
final
def
toPMessage: PMessage
- Definition Classes
- GeneratedMessage
-
def
toProtoString: String
- Definition Classes
- KernelDef → GeneratedMessage
- val unknownFields: UnknownFieldSet
-
def
update(ms: (Lens[KernelDef, KernelDef]) ⇒ Mutation[KernelDef]*): KernelDef
- Definition Classes
- Updatable
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
- def withConstraint(__v: Seq[AttrConstraint]): KernelDef
- def withDeviceType(__v: String): KernelDef
- def withHostMemoryArg(__v: Seq[String]): KernelDef
- def withLabel(__v: String): KernelDef
- def withOp(__v: String): KernelDef
- def withPriority(__v: Int): KernelDef
- def withUnknownFields(__v: UnknownFieldSet): KernelDef
-
final
def
writeDelimitedTo(output: OutputStream): Unit
- Definition Classes
- GeneratedMessage
-
def
writeTo(_output__: CodedOutputStream): Unit
- Definition Classes
- KernelDef → GeneratedMessage
-
final
def
writeTo(output: OutputStream): Unit
- Definition Classes
- GeneratedMessage
Deprecated Value Members
-
def
finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] ) @Deprecated
- Deprecated