Struct tensorflow_proto::xla::gpu::CudnnConvBackendConfig

pub struct CudnnConvBackendConfig {
    pub algorithm: i64,
    pub tensor_ops_enabled: bool,
    pub conv_result_scale: f64,
    pub activation_mode: i64,
    pub side_input_scale: f64,
}

Backend config for a convolution that runs through cudnn.

Fields

algorithm: i64

Opaque algorithm number of cudnn algorithm chosen for this conv.

tensor_ops_enabled: bool

Whether we may use tensor cores when running this conv. Even if this is true, cudnn may choose not to use tensor cores, e.g. because the GPU or selected algorithm doesn't support it.

conv_result_scale: f64

The scaling factor multiplied with the convolution result.

activation_mode: i64

The requested activation (e.g. relu) after the convolution. It is with type stream_executor::dnn::ActivationMode.

side_input_scale: f64

The scaling factor multiplied with the side input. If no side input buffer is provided, this field must be 0.

Trait Implementations

impl Clone for CudnnConvBackendConfig

pub fn clone(&self) -> CudnnConvBackendConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for CudnnConvBackendConfig

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

Formats the value using the given formatter.

impl Default for CudnnConvBackendConfig

pub fn default() -> Self

Returns the "default value" for a type.

impl Message for CudnnConvBackendConfig

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<CudnnConvBackendConfig> for CudnnConvBackendConfig

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

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

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

This method tests for !=.

impl StructuralPartialEq for CudnnConvBackendConfig