Struct tensorflow::ops::SnapshotDataset

pub struct SnapshotDataset { /* private fields */ }

Builder for the SnapshotDataset operation.

Implementations

impl SnapshotDataset

pub fn new() -> Self

Creates a new SnapshotDataset.

pub fn output_types<ArgType: Into<Vec<DataType>>>(self, value: ArgType) -> Self

Sets the output_types attribute.

pub fn output_shapes<ArgType: Into<Vec<Shape>>>(self, value: ArgType) -> Self

Sets the output_shapes attribute.

pub fn compression<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the compression attribute.

pub fn reader_path_prefix<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the reader_path_prefix attribute.

pub fn writer_path_prefix<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the writer_path_prefix attribute.

pub fn shard_size_bytes<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the shard_size_bytes attribute.

pub fn pending_snapshot_expiry_seconds<ArgType: Into<i64>>( self, value: ArgType ) -> Self

Sets the pending_snapshot_expiry_seconds attribute.

pub fn num_reader_threads<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the num_reader_threads attribute.

pub fn reader_buffer_size<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the reader_buffer_size attribute.

pub fn num_writer_threads<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the num_writer_threads attribute.

pub fn writer_buffer_size<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the writer_buffer_size attribute.

pub fn shuffle_on_read<ArgType: Into<bool>>(self, value: ArgType) -> Self

Sets the shuffle_on_read attribute.

pub fn seed<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the seed attribute.

pub fn seed2<ArgType: Into<i64>>(self, value: ArgType) -> Self

Sets the seed2 attribute.

pub fn mode<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the mode attribute.

pub fn snapshot_name<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the snapshot_name attribute.

pub fn add_control_input(self, op: Operation) -> Self

Adds a control input.

pub fn build<O0: Into<Output>, O1: Into<Output>>( &self, input_dataset: O0, path: O1, scope: &mut Scope ) -> Result<Operation>

Builds the SnapshotDataset operation.

pub fn build_instance( &self, input_dataset: Output, path: Output, scope: &mut Scope ) -> Result<SnapshotDatasetInst>

Builds a new instance of ‘SnapshotDataset’ Operation with it’s Outputs and Inputs exposed as methods.

Trait Implementations

impl Debug for SnapshotDataset

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

Formats the value using the given formatter.

impl Default for SnapshotDataset

fn default() -> SnapshotDataset

Returns the “default value” for a type.