|
|
@@ -24,7 +24,6 @@ def conv_bias_activation( |
|
|
|
padding: Union[int, Tuple[int, int]] = 0, |
|
|
|
dilation: Union[int, Tuple[int, int]] = 1, |
|
|
|
groups: int = 1, |
|
|
|
format="NCHW", |
|
|
|
nonlinear_mode="IDENTITY", |
|
|
|
conv_mode="CROSS_CORRELATION", |
|
|
|
compute_mode="DEFAULT", |
|
|
@@ -69,7 +68,7 @@ def conv_bias_activation( |
|
|
|
dilate_h=dh, |
|
|
|
dilate_w=dw, |
|
|
|
dtype=dtype, |
|
|
|
format=format, |
|
|
|
format="NCHW", |
|
|
|
strategy=get_conv_execution_strategy(), |
|
|
|
nonlineMode=nonlinear_mode, |
|
|
|
mode=conv_mode, |
|
|
|