[−][src]Struct tract_tensorflow::tfpb::tensorflow::NodeDef
Fields
name: String
The name given to this operator. Used for naming inputs, logging, visualization, etc. Unique within a single GraphDef. Must match the regexp "[A-Za-z0-9.][A-Za-z0-9_./]*".
op: String
The operation name. There may be custom parameters in attrs. Op names starting with an underscore are reserved for internal use.
input: Vec<String>
Each input is "node:src_output" with "node" being a string name and "src_output" indicating which output tensor to use from "node". If "src_output" is 0 the ":0" suffix can be omitted. Regular inputs may optionally be followed by control inputs that have the format "^node".
device: String
A (possibly partial) specification for the device on which this node should be placed. The expected syntax for this string is as follows:
DEVICE_SPEC ::= PARTIAL_SPEC
PARTIAL_SPEC ::= ("/" CONSTRAINT) * CONSTRAINT ::= ("job:" JOB_NAME) | ("replica:" [1-9][0-9]) | ("task:" [1-9][0-9]) | ("device:" [A-Za-z]* ":" ([1-9][0-9]* | "*") )
Valid values for this string include:
- "/job:worker/replica:0/task:1/device:GPU:3" (full specification)
- "/job:worker/device:GPU:3" (partial specification)
- "" (no specification)
If the constraints do not resolve to a single device (or if this field is empty or not present), the runtime will attempt to choose a device automatically.
attr: HashMap<String, AttrValue>
Operation-specific graph-construction-time configuration. Note that this should include all attrs defined in the corresponding OpDef, including those with a value matching the default -- this allows the default to change and makes NodeDefs easier to interpret on their own. However, if an attr with a default is not specified in this list, the default will be used. The "names" (keys) must match the regexp "[a-z][a-z0-9_]+" (and one of the names from the corresponding OpDef's attr field). The values must have a type matching the corresponding OpDef attr's type field. TODO(josh11b): Add some examples here showing best practices.
Implementations
impl NodeDef
[src]
pub fn name<S: ToString>(mut self: Self, n: S) -> NodeDef
[src]
pub fn op<S: ToString>(mut self: Self, n: S) -> NodeDef
[src]
pub fn input<S: ToString>(mut self: Self, n: S) -> NodeDef
[src]
pub fn attr<S: ToString, V: Into<AttrValue>>(
mut self: Self,
n: S,
v: V
) -> NodeDef
[src]
mut self: Self,
n: S,
v: V
) -> NodeDef
impl NodeDef
[src]
pub fn get_attr_raw_str(&self, name: &str) -> TractResult<&[u8]>
[src]
pub fn get_attr_opt_raw_str(&self, name: &str) -> TractResult<Option<&[u8]>>
[src]
pub fn get_attr_str(&self, name: &str) -> TractResult<String>
[src]
pub fn get_attr_opt_str(&self, name: &str) -> TractResult<Option<String>>
[src]
pub fn get_attr_bool(&self, name: &str) -> TractResult<bool>
[src]
pub fn get_attr_opt_bool(&self, name: &str) -> TractResult<Option<bool>>
[src]
pub fn get_attr_datum_type(&self, name: &str) -> TractResult<DatumType>
[src]
pub fn get_attr_opt_datum_type(
&self,
name: &str
) -> TractResult<Option<DatumType>>
[src]
&self,
name: &str
) -> TractResult<Option<DatumType>>
pub fn get_attr_shape(&self, name: &str) -> TractResult<TVec<isize>>
[src]
pub fn get_attr_opt_shape(&self, name: &str) -> TractResult<Option<TVec<isize>>>
[src]
pub fn get_attr_tensor(&self, name: &str) -> TractResult<Tensor>
[src]
pub fn get_attr_opt_tensor(&self, name: &str) -> TractResult<Option<Tensor>>
[src]
pub fn get_attr_int<T: FromPrimitive>(&self, name: &str) -> TractResult<T>
[src]
pub fn get_attr_opt_int<T: FromPrimitive>(
&self,
name: &str
) -> TractResult<Option<T>>
[src]
&self,
name: &str
) -> TractResult<Option<T>>
pub fn get_attr_float<T: FromPrimitive>(&self, name: &str) -> TractResult<T>
[src]
pub fn get_attr_opt_float<T: FromPrimitive>(
&self,
name: &str
) -> TractResult<Option<T>>
[src]
&self,
name: &str
) -> TractResult<Option<T>>
pub fn get_attr_list_int<T: FromPrimitive>(
&self,
name: &str
) -> TractResult<Vec<T>>
[src]
&self,
name: &str
) -> TractResult<Vec<T>>
pub fn get_attr_opt_list_int<T: FromPrimitive>(
&self,
name: &str
) -> TractResult<Option<Vec<T>>>
[src]
&self,
name: &str
) -> TractResult<Option<Vec<T>>>
Trait Implementations
impl Clone for NodeDef
[src]
impl Debug for NodeDef
[src]
impl Default for NodeDef
[src]
impl Message for NodeDef
[src]
pub fn encode_raw<B>(&self, buf: &mut B) where
B: BufMut,
[src]
B: BufMut,
pub fn merge_field<B>(
&mut self,
tag: u32,
wire_type: WireType,
buf: &mut B,
ctx: DecodeContext
) -> Result<(), DecodeError> where
B: Buf,
[src]
&mut self,
tag: u32,
wire_type: WireType,
buf: &mut B,
ctx: DecodeContext
) -> Result<(), DecodeError> where
B: Buf,
pub fn encoded_len(&self) -> usize
[src]
pub fn clear(&mut self)
[src]
pub fn encode<B>(&self, buf: &mut B) -> Result<(), EncodeError> where
B: BufMut,
[src]
B: BufMut,
pub fn encode_length_delimited<B>(&self, buf: &mut B) -> Result<(), EncodeError> where
B: BufMut,
[src]
B: BufMut,
pub fn decode<B>(buf: B) -> Result<Self, DecodeError> where
Self: Default,
B: Buf,
[src]
Self: Default,
B: Buf,
pub fn decode_length_delimited<B>(buf: B) -> Result<Self, DecodeError> where
Self: Default,
B: Buf,
[src]
Self: Default,
B: Buf,
pub fn merge<B>(&mut self, buf: B) -> Result<(), DecodeError> where
B: Buf,
[src]
B: Buf,
pub fn merge_length_delimited<B>(&mut self, buf: B) -> Result<(), DecodeError> where
B: Buf,
[src]
B: Buf,
impl PartialEq<NodeDef> for NodeDef
[src]
impl StructuralPartialEq for NodeDef
[src]
Auto Trait Implementations
impl RefUnwindSafe for NodeDef
[src]
impl Send for NodeDef
[src]
impl Sync for NodeDef
[src]
impl Unpin for NodeDef
[src]
impl UnwindSafe for NodeDef
[src]
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized,
[src]
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
pub fn borrow_mut(&mut self) -> &mut Tⓘ
[src]
impl<T> Downcast for T where
T: Any,
T: Any,
pub fn into_any(self: Box<T, Global>) -> Box<dyn Any + 'static, Global>ⓘNotable traits for Box<R, Global>
impl<R> Read for Box<R, Global> where
R: Read + ?Sized, impl<W> Write for Box<W, Global> where
W: Write + ?Sized, impl<F, A> Future for Box<F, A> where
F: Unpin + Future + ?Sized,
A: Allocator + 'static, type Output = <F as Future>::Output;impl<I, A> Iterator for Box<I, A> where
I: Iterator + ?Sized,
A: Allocator, type Item = <I as Iterator>::Item;
Notable traits for Box<R, Global>
impl<R> Read for Box<R, Global> where
R: Read + ?Sized, impl<W> Write for Box<W, Global> where
W: Write + ?Sized, impl<F, A> Future for Box<F, A> where
F: Unpin + Future + ?Sized,
A: Allocator + 'static, type Output = <F as Future>::Output;impl<I, A> Iterator for Box<I, A> where
I: Iterator + ?Sized,
A: Allocator, type Item = <I as Iterator>::Item;
pub fn into_any_rc(self: Rc<T>) -> Rc<dyn Any + 'static>
pub fn as_any(&self) -> &(dyn Any + 'static)
pub fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
impl<T> DowncastSync for T where
T: Send + Sync + Any,
T: Send + Sync + Any,
impl<T> DynClone for T where
T: Clone,
[src]
T: Clone,
pub fn __clone_box(&self, Private) -> *mut ()
[src]
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<T> ToOwned for T where
T: Clone,
[src]
T: Clone,
type Owned = T
The resulting type after obtaining ownership.
pub fn to_owned(&self) -> T
[src]
pub fn clone_into(&self, target: &mut T)
[src]
impl<T, U> TryFrom<U> for T where
U: Into<T>,
[src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
pub fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>
[src]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,