Struct tfrecord::protobuf::op_def::ArgDef

pub struct ArgDef {
    pub name: String,
    pub description: String,
    pub type: i32,
    pub type_attr: String,
    pub number_attr: String,
    pub type_list_attr: String,
    pub handle_data: Vec<DtypeAndShape>,
    pub is_ref: bool,
    pub experimental_full_type: Option<FullTypeDef>,
}

For describing inputs and outputs.

Fields

name: String

Name for the input/output. Should match the regexp “[a-z][a-z0-9_]*”.

description: String

Human readable description.

type: i32

Describes the type of one or more tensors that are accepted/produced by this input/output arg. The only legal combinations are:

• For a single tensor: either the “type” field is set or the “type_attr” field is set to the name of an attr with type “type”.
• For a sequence of tensors with the same type: the “number_attr” field will be set to the name of an attr with type “int”, and either the “type” or “type_attr” field will be set as for single tensors.
• For a sequence of tensors, the “type_list_attr” field will be set to the name of an attr with type “list(type)”.

type_attr: String

if specified, attr must have type “type”

number_attr: String

if specified, attr must have type “int”

type_list_attr: String

If specified, attr must have type “list(type)”, and none of type, type_attr, and number_attr may be specified.

handle_data: Vec<DtypeAndShape>

The handle data for resource inputs.

is_ref: bool

For inputs: if true, the inputs are required to be refs. By default, inputs can be either refs or non-refs. For outputs: if true, outputs are refs, otherwise they are not.

experimental_full_type: Option<FullTypeDef>

Experimental. Full type declaration for this argument. The full type specification combines type, type_attr, type_list_attr, etc. into a unified representation. This declaration may contain non-concrete types (for example, Tensor<TypeVar<‘T’>> is a valid type declaration.

Note: this is a transient field. The long-term aim is to represent the entire OpDef as a single type: a callable. In that context, this field is just the type of a single argument.

Implementations

impl ArgDef

pub fn type(&self) -> DataType

Returns the enum value of type, or the default if the field is set to an invalid enum value.

pub fn set_type(&mut self, value: DataType)

Sets type to the provided enum value.

Trait Implementations

impl Clone for ArgDef

fn clone(&self) -> ArgDef

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ArgDef

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

Formats the value using the given formatter.

impl Default for ArgDef

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ArgDef

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

Deserialize this value from the given Serde deserializer.

impl Message for ArgDef

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.

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

Encodes the message to a buffer.

fn encode_to_vec(&self) -> Vec<u8, Global>

Encodes the message to a newly allocated buffer.

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

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

fn encode_length_delimited_to_vec(&self) -> Vec<u8, Global>

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

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

Decodes an instance of the message from a buffer.

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

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

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.

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<ArgDef> for ArgDef

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

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

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

This method tests for !=.

impl Serialize for ArgDef

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 ArgDef