@Namespace(value="dnnl") public static enum dnnl.algorithm extends Enum<dnnl.algorithm>
| Enum Constant and Description |
|---|
binary_add
Binary add
|
binary_max
Binary max
|
binary_min
Binary min
|
binary_mul
Binary mul
|
convolution_auto
Convolution algorithm that is chosen to be either direct or Winograd
automatically
|
convolution_direct
Direct convolution
|
convolution_winograd
Winograd convolution
|
deconvolution_direct
Direct deconvolution
|
deconvolution_winograd
Winograd deconvolution
|
eltwise_abs
Elementwise: abs
|
eltwise_bounded_relu
Elementwise: bounded_relu
|
eltwise_clip
Elementwise: clip
|
eltwise_elu
Elementwise: exponential linear unit (ELU)
|
eltwise_elu_use_dst_for_bwd
Elementwise: exponential linear unit (ELU) (dst for backward)
|
eltwise_exp
Elementwise: exponent
|
eltwise_exp_use_dst_for_bwd
Elementwise: exponent (dst for backward)
|
eltwise_gelu
Elementwise: gelu
alias for #dnnl::algorithm::eltwise_gelu_tanh
|
eltwise_gelu_erf
Elementwise: erf-based gelu
|
eltwise_gelu_tanh
Elementwise: tanh-based gelu
|
eltwise_linear
Elementwise: linear
|
eltwise_log
Elementwise: natural logarithm
|
eltwise_logistic
Elementwise: logistic
|
eltwise_logistic_use_dst_for_bwd
Elementwise: logistic (dst for backward)
|
eltwise_pow
Elementwise: pow
|
eltwise_relu
Elementwise: rectified linear unit (ReLU)
|
eltwise_relu_use_dst_for_bwd
Elementwise: rectified linar unit (ReLU) (dst for backward)
|
eltwise_round
Elementwise: round
|
eltwise_soft_relu
Elementwise: soft_relu
|
eltwise_sqrt
Elementwise: square root
|
eltwise_sqrt_use_dst_for_bwd
Elementwise: square root (dst for backward)
|
eltwise_square
Elementwise: square
|
eltwise_swish
Elementwise: swish (
x \cdot sigmoid(a \cdot x)) |
eltwise_tanh
Elementwise: hyperbolic tangent non-linearity (tanh)
|
eltwise_tanh_use_dst_for_bwd
Elementwise: hyperbolic tangent non-linearity (tanh) (dst for backward)
|
lbr_gru
GRU cell with linear before reset.
|
lrn_across_channels
Local response normalization (LRN) across multiple channels
|
lrn_within_channel
LRN within a single channel
|
pooling_avg
Average pooling exclude padding,
alias for #dnnl::algorithm::pooling_avg_include_padding
|
pooling_avg_exclude_padding
Average pooling exclude padding
|
pooling_avg_include_padding
Average pooling include padding
|
pooling_max
Max pooling
|
resampling_linear
Linear (Bilinear, Trilinear) resampling method
|
resampling_nearest
Nearest Neighbor resampling method
|
undef
Undefined algorithm
|
vanilla_gru
GRU cell
|
vanilla_lstm
LSTM cell
|
vanilla_rnn
RNN cell
|
| Modifier and Type | Field and Description |
|---|---|
int |
value |
| Modifier and Type | Method and Description |
|---|---|
dnnl.algorithm |
intern() |
String |
toString() |
static dnnl.algorithm |
valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static dnnl.algorithm[] |
values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
public static final dnnl.algorithm undef
public static final dnnl.algorithm convolution_auto
public static final dnnl.algorithm convolution_direct
public static final dnnl.algorithm convolution_winograd
public static final dnnl.algorithm deconvolution_direct
public static final dnnl.algorithm deconvolution_winograd
public static final dnnl.algorithm eltwise_relu
public static final dnnl.algorithm eltwise_tanh
public static final dnnl.algorithm eltwise_elu
public static final dnnl.algorithm eltwise_square
public static final dnnl.algorithm eltwise_abs
public static final dnnl.algorithm eltwise_sqrt
public static final dnnl.algorithm eltwise_swish
x \cdot sigmoid(a \cdot x))public static final dnnl.algorithm eltwise_linear
public static final dnnl.algorithm eltwise_bounded_relu
public static final dnnl.algorithm eltwise_soft_relu
public static final dnnl.algorithm eltwise_logistic
public static final dnnl.algorithm eltwise_exp
public static final dnnl.algorithm eltwise_gelu
public static final dnnl.algorithm eltwise_gelu_tanh
public static final dnnl.algorithm eltwise_gelu_erf
public static final dnnl.algorithm eltwise_log
public static final dnnl.algorithm eltwise_clip
public static final dnnl.algorithm eltwise_pow
public static final dnnl.algorithm eltwise_round
public static final dnnl.algorithm eltwise_relu_use_dst_for_bwd
public static final dnnl.algorithm eltwise_tanh_use_dst_for_bwd
public static final dnnl.algorithm eltwise_elu_use_dst_for_bwd
public static final dnnl.algorithm eltwise_sqrt_use_dst_for_bwd
public static final dnnl.algorithm eltwise_logistic_use_dst_for_bwd
public static final dnnl.algorithm eltwise_exp_use_dst_for_bwd
public static final dnnl.algorithm lrn_across_channels
public static final dnnl.algorithm lrn_within_channel
public static final dnnl.algorithm pooling_max
public static final dnnl.algorithm pooling_avg
public static final dnnl.algorithm pooling_avg_include_padding
public static final dnnl.algorithm pooling_avg_exclude_padding
public static final dnnl.algorithm vanilla_rnn
public static final dnnl.algorithm vanilla_lstm
public static final dnnl.algorithm vanilla_gru
public static final dnnl.algorithm lbr_gru
c_t = tanh(W_c*x_t + b_{c_x} + r_t*(U_c*h_{t-1}+b_{c_h}))
LRB GRU expects 4 bias tensors on input:
[b_{u}, b_{r}, b_{c_x}, b_{c_h}]public static final dnnl.algorithm binary_add
public static final dnnl.algorithm binary_mul
public static final dnnl.algorithm binary_max
public static final dnnl.algorithm binary_min
public static final dnnl.algorithm resampling_nearest
public static final dnnl.algorithm resampling_linear
public static dnnl.algorithm[] values()
for (dnnl.algorithm c : dnnl.algorithm.values()) System.out.println(c);
public static dnnl.algorithm valueOf(String name)
name - the name of the enum constant to be returned.IllegalArgumentException - if this enum type has no constant with the specified nameNullPointerException - if the argument is nullpublic dnnl.algorithm intern()
public String toString()
toString in class Enum<dnnl.algorithm>Copyright © 2020. All rights reserved.