Struct tfrecord::protos::KernelDef

pub struct KernelDef {
    pub op: String,
    pub device_type: String,
    pub constraint: Vec<AttrConstraint>,
    pub host_memory_arg: Vec<String>,
    pub label: String,
    pub priority: i32,
}

Fields

op: String

Must match the name of an Op.

device_type: String

Type of device this kernel runs on.

constraint: Vec<AttrConstraint>

host_memory_arg: Vec<String>

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

label: String

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: i32

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.

Trait Implementations

impl Clone for KernelDef

fn clone(&self) -> KernelDef

Returns a copy of the value.

pub fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for KernelDef

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for KernelDef

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for KernelDef

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
    __D: Deserializer<'de>, 

Deserialize this value from the given Serde deserializer.

impl Message for KernelDef

fn encode_raw<B>(&self, buf: &mut B) where
    B: BufMut, 

fn merge_field<B>(
    &mut self,
    tag: u32,
    wire_type: WireType,
    buf: &mut B,
    ctx: DecodeContext
) -> Result<(), DecodeError> where
    B: Buf, 

fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

fn clear(&mut self)

Clears the message, resetting all fields to their default.

pub fn encode<B>(&self, buf: &mut B) -> Result<(), EncodeError> where
    B: BufMut, 

Encodes the message to a buffer.

pub fn encode_length_delimited<B>(&self, buf: &mut B) -> Result<(), EncodeError> where
    B: BufMut, 

Encodes the message with a length-delimiter to a buffer.

pub fn decode<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

Decodes an instance of the message from a buffer.

pub fn decode_length_delimited<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

Decodes a length-delimited instance of the message from the buffer.

pub fn merge<B>(&mut self, buf: B) -> Result<(), DecodeError> where
    B: Buf, 

Decodes an instance of the message from a buffer, and merges it into self.

pub fn merge_length_delimited<B>(&mut self, buf: B) -> Result<(), DecodeError> where
    B: Buf, 

Decodes a length-delimited instance of the message from buffer, and merges it into self.

impl PartialEq<KernelDef> for KernelDef

fn eq(&self, other: &KernelDef) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &KernelDef) -> bool

This method tests for !=.

impl Serialize for KernelDef

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error> where
    __S: Serializer, 

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for KernelDef