Struct tract_tensorflow::tfpb::tensorflow::GraphDef

pub struct GraphDef {
    pub node: Vec<NodeDef>,
    pub versions: Option<VersionDef>,
    pub version: i32,
    pub library: Option<FunctionDefLibrary>,
}

Represents the graph of operations

Fields

node: Vec<NodeDef>

versions: Option<VersionDef>

Compatibility versions of the graph. See core/public/version.h for version history. The GraphDef version is distinct from the TensorFlow version, and each release of TensorFlow will support a range of GraphDef versions.

version: i32

Deprecated single version field; use versions above instead. Since all GraphDef changes before "versions" was introduced were forward compatible, this field is entirely ignored.

library: Option<FunctionDefLibrary>

EXPERIMENTAL. DO NOT USE OR DEPEND ON THIS YET.

"library" provides user-defined functions.

Naming:

• library.function.name are in a flat namespace. NOTE: We may need to change it to be hierarchical to support different orgs. E.g., { "/google/nn", { ... }}, { "/google/vision", { ... }} { "/org_foo/module_bar", { ... }} map<string, FunctionDefLib> named_lib;
• If node[i].op is the name of one function in "library", node[i] is deemed as a function call. Otherwise, node[i].op must be a primitive operation supported by the runtime.

Function call semantics:

• The callee may start execution as soon as some of its inputs are ready. The caller may want to use Tuple() mechanism to ensure all inputs are ready in the same time.

• The consumer of return values may start executing as soon as the return values the consumer depends on are ready. The consumer may want to use Tuple() mechanism to ensure the consumer does not start until all return values of the callee function are ready.

Methods

impl GraphDef

pub fn node(self, n: NodeDef) -> Self

pub fn write_to_bytes(&self) -> TractResult<Vec<u8>>

pub fn save_to<P: AsRef<Path>>(self, p: P) -> TractResult<()>

Trait Implementations

impl Clone for GraphDef

fn clone(&self) -> GraphDef

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for GraphDef

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

Formats the value using the given formatter.

impl Default for GraphDef

fn default() -> GraphDef

Returns the "default value" for a type.

impl Framework<GraphDef> for Tensorflow

fn proto_model_for_path(&self, r: impl AsRef<Path>) -> TractResult<GraphDef>

This method will try to read as frozen model, then as a saved model.

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

This method expects a frozen model, use open_saved_model for TF2 saved model format.

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

Translate a proto model into a model.

fn model_for_read(
    &self,
    r: &mut dyn Read
) -> Result<ModelImpl<InferenceFact, Box<dyn InferenceOp + 'static>>, TractError>

Read a model from a reader

fn model_for_path(
    &self,
    p: impl AsRef<Path>
) -> Result<ModelImpl<InferenceFact, Box<dyn InferenceOp + 'static>>, TractError>

Build a model from a filename.

impl Message for GraphDef

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.

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

Encodes the message to a 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 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<GraphDef> for GraphDef

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

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

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

This method tests for !=.

impl StructuralPartialEq for GraphDef