Struct ConvolutionDescriptorProto

pub struct ConvolutionDescriptorProto {
    pub paddings: Vec<i64>,
    pub strides: Vec<i64>,
    pub dilations: Vec<i64>,
    pub compute_mode: i32,
    pub group_count: i32,
    pub convolution_mode: i32,
    pub name: String,
}

Convolution-specific parameters.

Fields

paddings: Vec<i64>

strides: Vec<i64>

dilations: Vec<i64>

compute_mode: i32

The “accumulator” type. For example, use F32 as an accumulator for F16 convolutions. See cuDNN’s cudnnConvolutionMode_t.

group_count: i32

See cuDNN’s group count.

convolution_mode: i32

name: String

Tensorflow node name, same as in NodeDef, for debugging purposes.

Implementations

impl ConvolutionDescriptorProto

pub fn compute_mode(&self) -> DataType

Returns the enum value of compute_mode, or the default if the field is set to an invalid enum value.

pub fn set_compute_mode(&mut self, value: DataType)

Sets compute_mode to the provided enum value.

pub fn convolution_mode(&self) -> ConvolutionMode

Returns the enum value of convolution_mode, or the default if the field is set to an invalid enum value.

pub fn set_convolution_mode(&mut self, value: ConvolutionMode)

Sets convolution_mode to the provided enum value.

Trait Implementations

impl Clone for ConvolutionDescriptorProto

fn clone(&self) -> ConvolutionDescriptorProto

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ConvolutionDescriptorProto

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

Formats the value using the given formatter.

impl Default for ConvolutionDescriptorProto

fn default() -> Self

Returns the “default value” for a type.

impl Message for ConvolutionDescriptorProto

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 ConvolutionDescriptorProto

fn eq(&self, other: &ConvolutionDescriptorProto) -> 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 ConvolutionDescriptorProto