package xla.gpu

Get desktop application:
View/edit binary Protocol Buffers messages

Backend config for a convolution that runs through cudnn.

• int64 algorithm = 1

  Opaque algorithm number of cudnn algorithm chosen for this conv.

• bool tensor_ops_enabled = 2

  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.

• double conv_result_scale = 4

  The scaling factor multiplied with the convolution result.

• int64 activation_mode = 3

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

• double side_input_scale = 5

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