Struct GraphOptions

pub struct GraphOptions {
    pub enable_recv_scheduling: bool,
    pub optimizer_options: Option<OptimizerOptions>,
    pub build_cost_model: i64,
    pub build_cost_model_after: i64,
    pub infer_shapes: bool,
    pub place_pruned_graph: bool,
    pub enable_bfloat16_sendrecv: bool,
    pub timeline_step: i32,
    pub rewrite_options: Option<RewriterConfig>,
}

Fields

enable_recv_scheduling: bool

If true, use control flow to schedule the activation of Recv nodes. (Currently ignored.)

optimizer_options: Option<OptimizerOptions>

Options controlling how graph is optimized.

build_cost_model: i64

The number of steps to run before returning a cost model detailing the memory usage and performance of each node of the graph. 0 means no cost model.

build_cost_model_after: i64

The number of steps to skip before collecting statistics for the cost model.

infer_shapes: bool

Annotate each Node with Op output shape data, to the extent it can be statically inferred.

place_pruned_graph: bool

Only place the subgraphs that are run, rather than the entire graph.

This is useful for interactive graph building, where one might produce graphs that cannot be placed during the debugging process. In particular, it allows the client to continue work in a session after adding a node to a graph whose placement constraints are unsatisfiable.

enable_bfloat16_sendrecv: bool

If true, transfer float values between processes as bfloat16.

timeline_step: i32

If > 0, record a timeline every this many steps. EXPERIMENTAL: This currently has no effect in MasterSession.

rewrite_options: Option<RewriterConfig>

Options that control the type and amount of graph rewriting. Not currently configurable via the public Python API (i.e. there is no API stability guarantee if you import RewriterConfig explicitly).

Trait Implementations

impl Clone for GraphOptions

fn clone(&self) -> GraphOptions

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphOptions

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

Formats the value using the given formatter.

impl Default for GraphOptions

fn default() -> Self

Returns the “default value” for a type.

impl Message for GraphOptions

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, Self: Sized,

Encodes the message to a buffer.

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

where B: BufMut, Self: Sized,

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, Self: Sized,

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, Self: Sized,

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

impl PartialEq for GraphOptions

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for GraphOptions