Struct tensorflow_proto::stream_executor::dnn::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

pub fn clone(&self) -> ConvolutionDescriptorProto

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ConvolutionDescriptorProto

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

Formats the value using the given formatter.

impl Default for ConvolutionDescriptorProto

pub fn default() -> Self

Returns the "default value" for a type.

impl Message for ConvolutionDescriptorProto

pub fn encode_raw<B>(&self, buf: &mut B) where
    B: BufMut, 

pub fn merge_field<B>(
    &mut self,
    tag: u32,
    wire_type: WireType,
    buf: &mut B,
    ctx: DecodeContext
) -> Result<(), DecodeError> where
    B: Buf, 

pub fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

pub fn clear(&mut self)

Clears the message, resetting all fields to their default.

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

Encodes the message to a buffer.

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

Encodes the message with a length-delimiter to a buffer.

pub fn decode<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

Decodes an instance of the message from a buffer.

pub fn decode_length_delimited<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

Decodes a length-delimited instance of the message from the buffer.

pub 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.

pub 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<ConvolutionDescriptorProto> for ConvolutionDescriptorProto

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

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

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

This method tests for !=.

impl StructuralPartialEq for ConvolutionDescriptorProto