Struct tract_tensorflow::tfpb::node_def::NodeDef

pub struct NodeDef {
    pub name: String,
    pub op: String,
    pub input: RepeatedField<String>,
    pub device: String,
    pub attr: HashMap<String, AttrValue>,
    pub unknown_fields: UnknownFields,
    pub cached_size: CachedSize,
}

Fields

name: String

op: String

input: RepeatedField<String>

device: String

attr: HashMap<String, AttrValue>

unknown_fields: UnknownFields

cached_size: CachedSize

Methods

impl NodeDef

pub fn new() -> NodeDef

pub fn clear_name(&mut self)

pub fn set_name(&mut self, v: String)

pub fn mut_name(&mut self) -> &mut String

pub fn take_name(&mut self) -> String

pub fn get_name(&self) -> &str

pub fn clear_op(&mut self)

pub fn set_op(&mut self, v: String)

pub fn mut_op(&mut self) -> &mut String

pub fn take_op(&mut self) -> String

pub fn get_op(&self) -> &str

pub fn clear_input(&mut self)

pub fn set_input(&mut self, v: RepeatedField<String>)

pub fn mut_input(&mut self) -> &mut RepeatedField<String>

pub fn take_input(&mut self) -> RepeatedField<String>

pub fn get_input(&self) -> &[String]

pub fn clear_device(&mut self)

pub fn set_device(&mut self, v: String)

pub fn mut_device(&mut self) -> &mut String

pub fn take_device(&mut self) -> String

pub fn get_device(&self) -> &str

pub fn clear_attr(&mut self)

pub fn set_attr(&mut self, v: HashMap<String, AttrValue>)

pub fn mut_attr(&mut self) -> &mut HashMap<String, AttrValue>

pub fn take_attr(&mut self) -> HashMap<String, AttrValue>

pub fn get_attr(&self) -> &HashMap<String, AttrValue>

impl NodeDef

pub fn name<S: ToString>(self, n: S) -> NodeDef

pub fn op<S: ToString>(self, n: S) -> NodeDef

pub fn input<S: ToString>(self, n: S) -> NodeDef

pub fn attr<S: ToString, V: Into<AttrValue>>(self, n: S, v: V) -> NodeDef

impl NodeDef

pub fn get_attr_raw_str(&self, name: &str) -> TractResult<&[u8]>

pub fn get_attr_opt_raw_str(&self, name: &str) -> TractResult<Option<&[u8]>>

pub fn get_attr_str(&self, name: &str) -> TractResult<String>

pub fn get_attr_opt_str(&self, name: &str) -> TractResult<Option<String>>

pub fn get_attr_bool(&self, name: &str) -> TractResult<bool>

pub fn get_attr_opt_bool(&self, name: &str) -> TractResult<Option<bool>>

pub fn get_attr_datum_type(&self, name: &str) -> TractResult<DatumType>

pub fn get_attr_opt_datum_type(
    &self,
    name: &str
) -> TractResult<Option<DatumType>>

pub fn get_attr_shape(&self, name: &str) -> TractResult<TVec<usize>>

pub fn get_attr_opt_shape(&self, name: &str) -> TractResult<Option<TVec<usize>>>

pub fn get_attr_tensor(&self, name: &str) -> TractResult<Tensor>

pub fn get_attr_opt_tensor(&self, name: &str) -> TractResult<Option<Tensor>>

pub fn get_attr_int<T: FromPrimitive>(&self, name: &str) -> TractResult<T>

pub fn get_attr_opt_int<T: FromPrimitive>(
    &self,
    name: &str
) -> TractResult<Option<T>>

pub fn get_attr_float<T: FromPrimitive>(&self, name: &str) -> TractResult<T>

pub fn get_attr_opt_float<T: FromPrimitive>(
    &self,
    name: &str
) -> TractResult<Option<T>>

pub fn get_attr_list_int<T: FromPrimitive>(
    &self,
    name: &str
) -> TractResult<Vec<T>>

pub fn get_attr_opt_list_int<T: FromPrimitive>(
    &self,
    name: &str
) -> TractResult<Option<Vec<T>>>

Trait Implementations

impl PartialEq<NodeDef> for NodeDef

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

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

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

This method tests for !=.

impl Default for NodeDef

fn default() -> NodeDef

Returns the "default value" for a type.

impl Clone for NodeDef

fn clone(&self) -> NodeDef

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for NodeDef

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

Formats the value using the given formatter.

impl Message for NodeDef

fn is_initialized(&self) -> bool

True iff all required fields are initialized. Always returns true for protobuf 3.

fn merge_from(&mut self, is: &mut CodedInputStream) -> ProtobufResult<()>

Update this message object with fields read from given stream.

fn compute_size(&self) -> u32

Compute and cache size of this message and all nested messages

fn write_to_with_cached_sizes(
    &self,
    os: &mut CodedOutputStream
) -> ProtobufResult<()>

Write message to the stream.

fn get_cached_size(&self) -> u32

Get size previously computed by compute_size.

fn get_unknown_fields(&self) -> &UnknownFields

Get a reference to unknown fields.

fn mut_unknown_fields(&mut self) -> &mut UnknownFields

Get a mutable reference to unknown fields.

fn as_any(&self) -> &dyn Any

View self as Any.

fn as_any_mut(&mut self) -> &mut dyn Any

View self as mutable Any.

fn into_any(self: Box<Self>) -> Box<dyn Any>

Convert boxed self to boxed Any.

fn descriptor(&self) -> &'static MessageDescriptor

Message descriptor for this message, used for reflection.

fn new() -> NodeDef

Create an empty message object.

fn descriptor_static() -> &'static MessageDescriptor

Get message descriptor for message type.

fn default_instance() -> &'static NodeDef

Return a pointer to default immutable message.

fn write_to(&self, os: &mut CodedOutputStream) -> Result<(), ProtobufError>

Write the message to the stream.

fn write_length_delimited_to(
    &self,
    os: &mut CodedOutputStream
) -> Result<(), ProtobufError>

Write the message to the stream prepending the message with message length encoded as varint.

fn write_length_delimited_to_vec(
    &self,
    vec: &mut Vec<u8>
) -> Result<(), ProtobufError>

Write the message to the vec, prepend the message with message length encoded as varint.

fn merge_from_bytes(&mut self, bytes: &[u8]) -> Result<(), ProtobufError>

Update this message object with fields read from given stream.

fn check_initialized(&self) -> Result<(), ProtobufError>

Check if all required fields of this object are initialized.

fn write_to_writer(&self, w: &mut dyn Write) -> Result<(), ProtobufError>

Write the message to the writer.

fn write_to_vec(&self, v: &mut Vec<u8>) -> Result<(), ProtobufError>

Write the message to bytes vec.

fn write_to_bytes(&self) -> Result<Vec<u8>, ProtobufError>

Write the message to bytes vec.

fn write_length_delimited_to_writer(
    &self,
    w: &mut dyn Write
) -> Result<(), ProtobufError>

Write the message to the writer, prepend the message with message length encoded as varint.

fn write_length_delimited_to_bytes(&self) -> Result<Vec<u8>, ProtobufError>

Write the message to the bytes vec, prepend the message with message length encoded as varint.

fn type_id(&self) -> TypeId

Get type id for downcasting.

impl Clear for NodeDef

fn clear(&mut self)

Clear this make, make it equivalent to newly created object.

impl ProtobufValue for NodeDef

fn as_ref(&self) -> ProtobufValueRef

fn as_any(&self) -> &(dyn Any + 'static)

fn is_non_zero(&self) -> bool

fn as_ref_copy(&self) -> ProtobufValueRef<'static>

impl Framework<NodeDef, GraphDef> for Tensorflow

fn op_builder_for_name(&self, name: &str) -> Option<&OpBuilder<NodeDef>>

fn proto_model_for_read(&self, r: &mut dyn Read) -> TractResult<GraphDef>

fn model_for_proto_model(&self, graph: &GraphDef) -> TractResult<InferenceModel>

fn proto_model_for_path<impl AsRef>(
    &self,
    p: impl AsRef
) -> Result<ProtoModel, TractError> where
    impl AsRef: AsRef<Path>, 

fn model_for_read(
    &self,
    r: &mut dyn Read
) -> Result<Model<TensorFact>, TractError>

fn model_for_path<impl AsRef>(
    &self,
    p: impl AsRef
) -> Result<Model<TensorFact>, TractError> where
    impl AsRef: AsRef<Path>, 

fn build_op(
    &self,
    name: &str,
    payload: &ProtoOp
) -> Result<Box<dyn Op + 'static>, TractError>